2811:熄灯问题

2811:熄灯问题
查看提交统计提示提问
总时间限制: 1000ms 内存限制: 65536kB
描述
有一个由按钮组成的矩阵，其中每行有6个按钮，共5行。每个按钮的位置上有一盏灯。当按下一个按钮后，该按钮以及周围位置(上边、下边、左边、右边)的灯都会改变一次。即，如果灯原来是点亮的，就会被熄灭；如果灯原来是熄灭的，则会被点亮。在矩阵角上的按钮改变3盏灯的状态；在矩阵边上的按钮改变4盏灯的状态；其他的按钮改变5盏灯的状态。

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

请你写一个程序，确定需要按下哪些按钮，恰好使得所有的灯都熄灭。根据上面的规则，我们知道1）第2次按下同一个按钮时，将抵消第1次按下时所产生的结果。因此，每个按钮最多只需要按下一次；2）各个按钮被按下的顺序对最终的结果没有影响；3）对第1行中每盏点亮的灯，按下第2行对应的按钮，就可以熄灭第1行的全部灯。如此重复下去，可以熄灭第1、2、3、4行的全部灯。同样，按下第1、2、3、4、5列的按钮，可以熄灭前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
来源
1222

/*
2811:熄灯问题
思路，如果枚举每一种开关状态需要2^30次尝试，故而考虑局部性，枚举第一行
开关从而确认后方每一行开关状态，注意位运算的运用 
*/
#include<cstring>
#include<iostream>
using namespace std;

char ori[8];	//原始灯矩阵 
char light[8];	//变化后的灯矩阵
char res[8];		//变化后最后的结果	

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(char result[])
{
	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()
{
	char switchs;	//记录某一行的开关状态 
	memset(ori,0,sizeof(ori));
	//初始化ori
	for(int i = 0;i < 5;++i)
	{
		for(int j = 0;j < 6;++j)
		{
			int k;
			cin >> k;
			setBit(ori[i],j,k);
		}
	} 
	

	//尝试第一行的64种开关方案可行性
	for(int k = 0;k < 64;++k)
	{
		memcpy(light,ori,sizeof(ori));
		switchs = k;
		for(int i = 0;i < 5;++i)
		{
			res[i] = switchs;
			for(int j = 0;j < 6;++j)
			{
				if(getBit(switchs,j))
				{
					if(j > 0)	Flip(light[i],j - 1);//改左边的灯 
					Flip(light[i],j);//改该位置的灯
					if(j < 5)	Flip(light[i],j + 1);	//改右灯 
				}
			}
			if(i < 4)	light[i + 1] ^= switchs;	//改下一行的灯  
			switchs = light[i];//下一行开关方案与本行相同 
		}
		//如果最后一行全0，则代表该方案正确
		if(light[4] == 0)
		{
			OutputResult(res);
			break;
		}
	}//for(int k = 0;k < 64;++k)
	
	 return 0;
}