Sample Input
2
0 1 1 0 1 0
1 0 0 1 1 1
0 0 1 0 0 1
1 0 0 1 0 1
0 1 1 1 0 0
0 0 1 0 1 0
1 0 1 0 1 1
0 0 1 0 1 1
1 0 1 1 0 0
0 1 0 1 0 0
Sample Output
PUZZLE #1
1 0 1 0 0 1
1 1 0 1 0 1
0 0 1 0 1 1
1 0 0 1 0 0
0 1 0 0 0 0
PUZZLE #2
1 0 0 1 1 1
1 1 0 0 0 0
0 0 0 1 0 0
1 1 0 1 0 1
1 0 1 1 0 1
Source
//POJ1222
#include<memory>
#include<string>
#include<cstring>
#include<iostream>
using namespace std;
char oriLights[5];//保存原始的5行,每个元素都有8位,足够对付每行的6个元素(0/1)
char lights[5];//变化中的灯
char result[5];//保存结果,开关的选择
int GetBit(char c,int i){//操作某个字符里的某个比特
return (c>>i)&1; //得到字符c的第i个比特
}
void SetBit(char &c,int i,int v){//把字符c的第i位设置为v(0/1),采用引用&,使得值会改变 (改变形参,实参也跟着变)
if(v){//v==1
c|=(1<<i);//把c的第i位变成1 |:见1变1见0不变
} else {//v==0
c&=~(1<<i);//把c的第i位变成0 &:见0变0见1不变
}
}
void FlipBit(char &c,int i){//翻转c的第i位
c^=(1<<i);//^:见1翻转 见0不变 1^(1/0)=(0/1) 0^(1/0)=(1/0)
}
void OutputResult(int t,char result[]){//输出结果,t代表第t个testcase
cout<<"PUZZLE #"<<t<<endl;
for(int i=0;i<5;++i){//5行
for(int j=0;j<6;++j){//6列
cout<<GetBit(result[i],j);//[i,j]对应result[i]的第j个比特
if(j<5) cout<<" ";
}
cout<<endl;//每输出一行后换行
}
}
int main(){
int T;
cin>>T;
for(int t=1;t<=T;++t){//对每个测试矩阵进行处理
for(int i=0;i<5;i++){
for(int j=0;j<6;j++){
int s;
cin>>s;
SetBit(oriLights[i],j,s);//把 oriLights[i]的第j比特换为s(0/1)
}
}//完成每组测试的输入
for(int n=0;n<64;++n){//每一个数n都对应一个第一行状态
int switchs=n;//代表当前行的开关灯状态
memcpy(lights,oriLights,sizeof(oriLights));
for(int i=0;i<5;++i){
result[i]=switchs;//保存了第i行的结果
for(int j=0;j<6;j++){
if(GetBit(switchs,j)){//如果是1(按了)的话翻转,是0的话不管
FlipBit(lights[i],j);
if(j>0)FlipBit(lights[i],j-1);
if(j<5)FlipBit(lights[i],j+1);
}
}//完成对第i行灯进行处理
if(i<4){
lights[i+1]^=switchs;//翻转第i+1行,此时i行以下都是不变的
}
switchs=lights[i];
} //以上作用之后可以确保前4行的灯都是灭的
if(lights[4]==0){//如果第5行全是关着的,输出并跳出循环。
OutputResult(t,result);
break;
}
}
}
return 0;
}