程序设计与算法（二）算法基础1.4熄灯问题

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

思路：

1.灯只有开和关两种状态，因此可以用比特串来表示。

2.当前n行灯的开关顺序固定时，第n+1行也固定（递归原理）

3.根据2，只要确定第一层的按灯的方式，整个矩阵就能确定，只要查看最后是否满足条件即可。

比特串的表示：

1.开一个5字节的数组，每个字节有八个比特，取六个代表一行的灯；每个字节代表一行。

2.第一行的按灯方式一共有64种，从0到63遍历即可。

技巧点：

1.关于位运算：

（1）：得到某一位：右移后与1作与；

（2）：某位变为1：1左移后作或；

（3）：某位变为0：1左移后取反，再作与；

（4）：反转某位：1左移后作异或。

2.第n行灯按完后，下一行灯的情况可以直接与第n行的按键作异或

3.第n行灯按完后，第n行的亮灯情况，就是下一行的按键情况（例如第4盏灯亮，则下一行第四个灯就需要按）。

#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#include<string.h>
char orilight[5];//每一行有8个比特，取其中6个用于表示
char light[5];
char result[5];
int gitbit(char c,int i)//得到第i位的比特
{
	return (c>>i)&1;
}
void setbit(char &c,int i,int v)//把第i位的比特变为v
{
	if(v){
		c|=(1<<i);
	}
	else{
		c&=~(1<<i);
	}
}
void flipbit(char &c,int i)//将第i位反转
{
	c^=(1<<i);
}
void output(int t,char result[])//输出结果
{
	printf("PUZZLE #%d\n",t+1);
	for(int i=0;i<5;i++){
		for(int j=0;j<6;j++){
			printf("%d ",gitbit(result[i],j));
		}
		printf("\n");
	}
	printf("\n");
}
int main()
{
	int T;
	scanf("%d",&T);
	for(int t=0;t<T;t++){//输入数据
		for(int i=0;i<5;i++){
			for(int j=0;j<6;j++){
				int a;
				scanf("%d",&a);
				setbit(orilight[i],j,a);
			}
		}
		for(int n=0;n<64;n++){//从2^0到2^6依次枚举
			int switchs=n;
			memcpy(light,orilight,sizeof(orilight));
			for(int i=0;i<5;i++){
				result[i]=switchs;//存结果
				for(int j=0;j<6;j++){//对同一行的灯产生影响
					if(gitbit(switchs,j)){
						if(j>0)
							flipbit(light[i],j-1);
						flipbit(light[i],j);
						if(j<5)
							flipbit(light[i],j+1);
					}
				}
				if(i<4)//对下一行的灯产生影响
					light[i+1]^=switchs;
				switchs=light[i];//本行亮的灯是下一行的开关
			}
			if(light[4]==0){
				output(t,result);
				break;
			}
		}
	}
	return 0;
}