传送门:http://poj.org/problem?id=1222
题意:开关灯问题,一幅开关的灯中,给出一种操作,使得灯全关掉,(操作一个开关,相邻的灯也会改变)
思路:利用位运算枚举第一行;
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
using namespace std; char orilights[],lights[],ans[]; int GetBit(char c, int i)
{
return (c>>i) & ;
}
void SetBit(char &c,int i,int v)
{
if(v)
{
c |= (<<i);
}
else
{
c &= ~(<<i);
}
}
void FlipBit(char &c, int i)
{
c ^= ( << i);
}
void Output(int t)
{
cout<<"PUZZLE #"<<t<<endl;
for(int i=; i<; ++i)
{
for(int j=; j<; ++j)
{
cout<< GetBit(ans[i],j);
if(j<)cout<<" ";
}
cout<<endl; }
}
int main(){
int T;
cin>>T;
for(int t = ; t<=T; t++)
{
for(int i=; i < ; ++i)
{
for(int j = ; j < ; ++j)
{
int s;
cin>>s;
SetBit(orilights[i], j, s);
}
}
//cout<<1<<endl;
for(int n=; n<; ++n)
{
int switchs = n;
memcpy(lights,orilights,sizeof(orilights));
for(int i=; i < ; ++i)
{
ans[i] = switchs;
for(int j = ;j < ; ++j)
{
if(GetBit(switchs,j))
{
if(j>)FlipBit(lights[i],j-);
FlipBit(lights[i],j);
if(j<)FlipBit(lights[i],j+);
}
}
if(i<)lights[i+] ^= switchs;
switchs = lights[i];
}
if(lights[]==)
{
Output(t);
break;
} }
}
return ;
}