POJ1222 EXTENDED LIGHTS OUT（熄灯问题）

点击这里查看原题

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;
}