POJ 2811:熄灯问题求指错

2811/1222

熄灯问题
描述
有一个由按钮组成的矩阵，其中每行有6个按钮，共5行。每个按钮的位置上有一盏灯。当按下一个按钮后，该按钮以及周围位置灯都会改变一次。即，如果灯原来是点亮的，就会被熄灭；如果灯原来是熄灭的，则会被点亮。在矩阵角上的按钮改变3盏灯的状态；在矩阵边上的按钮改变4盏灯的状态；其他的按钮改变5盏灯的状态。

在上图中，左边矩阵中用X标记的按钮表示被按下，右边的矩阵表示灯状态的改变。对矩阵中的每盏灯设置一个初始状态。请你按按钮，直至每一盏等都熄灭。与一盏灯毗邻的多个按钮被按下时，一个操作会抵消另一次操作的结果。在下图中，第2行第3、5列的按钮都被按下，因此第2行、第4列的灯的状态就不改变。

请你写一个程序，确定需要按下哪些按钮，恰好使得所有的灯都熄灭。根据上面的规则，我们知道
1）第2次按下同一个按钮时，将抵消第1次按下时所产生的结果。因此，每个按钮最多只需要按下一次；
2）各个按钮被按下的顺序对最终的结果没有影响；
3）对第1行中每盏点亮的灯，按下第2行对应的按钮，就可以熄灭第1行的全部灯。如此重复下去，可以熄灭第1234行的全部灯。同样，按下第12345列的按钮，可以熄灭前5列的灯。
输入
5行组成，每一行包括6个数字（0或1）。相邻两个数字之间用单个空格隔开。0表示灯的初始状态是熄灭的，1表示灯的初始状态是点亮的。
输出
5行组成，每一行包括6个数字（0或1）。相邻两个数字之间用单个空格隔开。其中的1表示需要把对应的按钮按下，0则表示不需要按对应的按钮。
样例输入
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
样例输出
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

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<memory> 
using namespace std;
char OriLights[5],Lights[5],Result[5];
void SetBit(char &p,int j,int v) //设置p的第j位为v  
{
    if(v) p|=(1<<j);
    else  p&=~(1<<j);
}
int GetBit(char p,int j)  //调用 
{
    return (p>>j)&1;
}
void FlipBit(char &p,int j)  // 将p的第j位取反 
{
    p^=(1<<j);
}
void Output(char p[])   //输出 
{
    for(int j=0;j<5;j++)  
    {
        for(int k=0;k<6;k++)
        {
            cout<<GetBit(p[j],k);
            if(k<5) cout<<' ';
        }
        if(j<4) cout<<"\n";
    }
}
int main()
{
        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++)      //操作 
        {
            int switchs=n;
            memcpy(Lights,OriLights,sizeof(OriLights));
            for(int i=0;i<5;++i)
            {
                Result[i]=switchs;
                for(int j=0;j<6;j++)
                {
                    if(GetBit(switchs,j))
                    {
                        if(j>0) FlipBit(Lights[i],j-1);
                        FlipBit(Lights[i],j);
                        if(j<5) FlipBit(Lights[i],j+1); 
                    }
                }
                if(i<4) Lights[i]^=switchs;
                switchs=Lights[i];
            }
            if(Lights[4]==0)
            {
                Output(Result);
                break;
            }
        }
    return 0;
}

用样例做输入，输出的跟样例不同

POJ 2811:熄灯问题 求指错

POJ 2811:熄灯问题求指错