1.有一個由按鈕組成的矩陣, 其中每行有6個按鈕, 共5行
2.每個按鈕的位置上有一盞燈
3.當按下一個按鈕后, 該按鈕以及周圍位置(上邊, 下邊, 左 邊, 右邊)的燈都會改變狀態
4.如果燈原來是點亮的, 就會被熄滅
5.如果燈原來是熄滅的, 則會被點亮
? 在矩陣角上的按鈕改變3盞燈的狀態
? 在矩陣邊上的按鈕改變4盞燈的狀態
? 其他的按鈕改變5盞燈的狀態
? 與一盞燈毗鄰的多個按鈕被按下時,一個操作會抵消另一次操 作的結果
? 給定矩陣中每盞燈的初始狀態，求一種按按鈕方案，使得所有的燈都熄滅
輸入
– 第一行是一個正整數N, 表示需要解決的案例數
– 每個案例由5行組成, 每一行包括6個數字
– 這些數字以空格隔開, 可以是0或1
– 0 表示燈的初始狀態是熄滅的
– 1 表示燈的初始狀態是點亮的
輸出
– 對每個案例, 首先輸出一行,輸出字符串 “PUZZLE #m”, 其中m是該案例的序號
– 接著按照該案例的輸入格式輸出5行
? 1 表示需要把對應的按鈕按下
? 0 表示不需要按對應的按鈕
? 每個數字以一個空格隔開
樣例輸入
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
樣例輸出
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

#include <memory>
#include <iostream>
#include <cstring>
using namespace std;

int GetBit(char c,int i){
    return (c>>i) & 1;
}
void SetBit(char &c,int i,int v){
    if(v){
        c|=(1<<i);
    }
    else{
        c&=~(1<<i);
    }
}
void Flip(char &c,int i){
    c^=(1<<i);
}
void OutputResult(int t,char result[]){
    cout<<"PUZZLE #"<<t<<endl;
    for(int i=0;i<5;++i){
        for(int j=0;j<6;++j){
            cout<<GetBit(result[i],j);
            if(j<5){
                cout<<" ";
            }
        }
        cout<<endl;
    }
}

int main(int argc, char const *argv[])
{
    char oriLights[5];
    char lights[5];
    char result[5];
    char switchs;
    int T;
    cin>>T;
    for(int t=1;t<=T;++t){
        memset(oriLights,0,sizeof(oriLights));
        for(int i=0;i<5;++i){
            for(int j=0;j<6;++j){
                int s;
                cin>>s;
                SetBit(oriLights[i],j,s);
            }
        }
        for(int n=0;n<64;++n){
            memcpy(lights,oriLights,sizeof(oriLights));
            switchs=n;
            for(int i=0;i<5;++i){
                result[i]=switchs;
                for(int j=0;j<6;++j){
                    if(GetBit(switchs,j)){
                        if(j>0){
                            Flip(lights[i],j-1);
                        }
                        Flip(lights[i],j);
                        if(j<5){
                            Flip(lights[i],j+1);
                        }
                    }
                }
                if(i<4){
                    lights[i+1]^=switchs;
                }
                switchs=lights[i];
            }
            if(lights[4]==0){
                OutputResult(t,result);
                break;
            }
        }
    }
    return 0;
}