利用位运算枚举 费解的开关

费解的开关

---

一、题目

你玩过“拉灯”游戏吗？25盏灯排成一个5x5的方形。每一个灯都有一个开关，游戏者可以改变它的状态。每一步，游戏者可以改变某一个灯的状态。游戏者改变一个灯的状态会产生连锁反应：和这个灯上下左右相邻的灯也要相应地改变其状态。
我们用数字“1”表示一盏开着的灯，用数字“0”表示关着的灯。下面这种状态
10111
01101
10111
10000
11011
在改变了最左上角的灯的状态后将变成：
01111
11101
10111
10000
11011
再改变它正中间的灯后状态将变成：
01111
11001
11001
10100
11011
给定一些游戏的初始状态，编写程序判断游戏者是否可能在6步以内使所有的灯都变亮。

二、输入

第一行有一个正整数n，代表数据中共有n个待解决的游戏初始状态。
以下若干行数据分为n组，每组数据有5行，每行5个字符。每组数据描述了一个游戏的初始状态。各组数据间用一个空行分隔。
对于30%的数据，n<=5；
对于100%的数据，n<=500。

三、输出

输出数据一共有n行，每行有一个小于等于6的整数，它表示对于输入数据中对应的游戏状态最少需要几步才能使所有灯变亮。
对于某一个游戏初始状态，若6步以内无法使所有灯变亮，请输出“-1”。

四、样例

样例一

input:

3
00111
01011
10001
11010
11100

11101
11101
11110
11111
11111

01111
11111
11111
11111
11111

output:

3
2
-1

五、思路和代码

5行灯泡对应5行开关，一旦第一行的开关确定了。后面各行的开关便确定了。后一行的开关必须保证上一行的灯全部开启，所以上一行的灯状态决定了下一行的开关状态。直到遍历到最后一行灯，如果最后一行灯符合要求，那么第一行的开关方案可行，否则不可行。

那么就要对第一行的开关进行枚举，有5位0和1，共有2⁵种可能，对应二进制00000B~11111B，可以利用位运算遍历，用一个字节的char型变量，从0到31进行枚举。char有8位，这里只需5位即可。

#include<iostream>
#include<cstring>
#define Min(a, b) a < b ? a : b
using namespace std;
char oriLights[5];//输入的灯泡矩阵
char lights[5];//计算中保存的灯泡矩阵
char result[5];//开关的结果矩阵
//取一个字符的第i个比特
int GetBit(char c, int i) {
	return 1 & (c >> i);
}
//将一个字符c的第i个比特设置为v
//用于输入初始的灯状态矩阵
void SetBit(char& c, int i, int v) {
	if (v == 1) {
		c |= (1 << i);
	}
	else {
		c &= ~(1 << i);
	}
}
//将字符c 的第i位翻转
//因为开关对灯泡有翻转的功能
void FlipBit(char& c, int i) {
	c ^= (1 << i);
}
//返回结果开关矩阵中1的数量 也就相当于执行的步数
int count(char result[]) {
	int res = 0;
	for (int i = 0; i < 5; i++) {
		for (int j = 0; j < 5; j++) {
			res += (result[i] >> j) & 1;
		}
	}
	return res;
}
int main() {
	char input[10];
	int t;
	cin >> t;
	while (t--) {
		int steps = 99999;
		//输入
		//题中1表示亮，这里0表示亮
		for (int i = 0; i < 5; i++) {
			cin >> input;
			for (int j = 0; j < 5; j++) {
				if (input[j] == '1')
				SetBit(oriLights[i], j, 0);
				if(input[j] == '0')
				SetBit(oriLights[i], j, 1);
			}
		}
		//遍历所有开关的排列情况
		for (int n = 0; n < 32; n++) {
			memcpy(lights, oriLights, sizeof(oriLights));
			//当前第0行开关状态
			int switches = n;
			//保存每行开关状态
			for (int i = 0; i < 5; i++) {
				//保存当前行的开关状态
				result[i] = switches;
				//对本行每个开关
				for (int j = 0; j < 5; j++) {
					if (GetBit(switches, j) == 1) {
						if (j > 0)
							FlipBit(lights[i], j - 1);
						FlipBit(lights[i], j);
						if (j < 4)
							FlipBit(lights[i], j + 1);
					}
				}
				if (i < 4) {
					//对下一行灯进行操作
					lights[i + 1] ^= switches;
				}
				//确定下一行的开关情况
				//为了让本行的灯全部开启，下一行的开关状态是唯一的
				switches = lights[i];
			}
			//最后一行灯是否全部开启
			if (lights[4] == 0) {
				steps = Min(steps, count(result));
			}
		}
		if (steps <= 6)
			cout << steps << endl;
		else
			cout << "-1" << endl;
	}
	return 0;
}