枚举与位运算——熄灯问题（c++)

声明：解法来自 北京大学 郭炜老师的程序设计与算法（二）算法基础

题目描述

右边)的灯都会改变一次。即，如果灯原来是点亮的，就会被熄灭；如果灯原来是熄灭的，则会被点亮。在矩阵角上的按钮改变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列的灯。

输入

T组数据，5行组成，每一行包括6个数字（0或1）。相邻两个数字之间用单个空格隔开。0表示灯的初始状态是熄灭的，1表示灯的初始状态是点亮的。

输出

5行组成，每一行包括6个数字（0或1）。相邻两个数字之间用单个空格隔开。其中的1表示需要把对应的按钮按下，0则表示不需要按对应的按钮。

样例输入

1
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

样例输出

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

代码

#include<iostream>
#include <string>
#include <cstring>
 
using namespace std;
char oriLights[5];
char lights[5];
char result[5];
int GetBit(char c,int i)	//得到第i位的值
{
	return (c>>i)&1;    //结果为1或0 
}
void SetBit(char & c,int i,int v){   //将c的第i位置为v 
	if(v){
		c |= ( 1 << i);    //将c的第i位置为1 
	}
	else
		c &= ~( 1 << i);	//将c的第i位置为0 
}
void FlipBit(char &c,int i)  //将c的第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 T;
	cin>>T;		//T组数据
	for(int t=1;t<=T;++t){
		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){	//共有64个可能性 
			int switchs=n;	
			memcpy(lights,oriLights,sizeof(oriLights)); 
			//将原始状态复制到lights[]中，后续操作在lights[]中进行 
			for(int i=0;i<5;++i){
				result[i]=switchs;	//第i行开关的状态 
				for(int j=0;j<6;j++){	//对第i行的灯进行处理 
					if(GetBit(switchs,j)){
						if(j>0)
							FlipBit(lights[i],j-1); //翻转j左侧的灯的状态 
						FlipBit(lights[i],j); 	//翻转j的状态 
						if(j<5)
							FlipBit(lights[i],j+1);	//翻转j右侧的灯的状态 
					}
				}
				if(i<4)
					lights[i+1] ^=switchs;  
					//如果switchs里的第i位是1，则lights[i+1]的第i位需要设为1 
				switchs=lights[i];
			}
			if(lights[4]==0){	//如果第四行全为0，则成功 
				OutputResult(t,result);	//输出第T组结果
				break;
			}
		}
	}
}

解题中用到的位运算

int GetBit(char c,int i)	//得到第i位的状态
{
	return (c>>i)&1;    //结果为1或0 
}

void SetBit(char & c,int i,int v){   //将c的第i位置为v 
	if(v){
		c |= ( 1 << i);    //将c的第i位置为1 
	}
	else
		c &= ~( 1 << i);	//将c的第i位置为0 
}

void FlipBit(char &c,int i)  //将c的第i位翻转 
{
	c ^= ( 1 << i);
}

左移运算符(>>)

c>>i
将c向左移动i位
定义：将一个运算对象的各二进制位全部左移若干位（左边的二进制位丢弃，右边补0）。
例如：设 令c=101101,i=4
则 c>>i 为000010
若左移时舍弃的高位不包含1，则每左移一位，相当于该数乘以2

右移运算符(<<)

定义：将一个数的各二进制位全部右移若干位，正数左补0，负数左补1，右边丢弃。
例如：设令c=101101,i=4
则 c<<i 为010000
操作数每右移一位，相当于该数除以2

按位与运算(&)

定义：参加运算的两个数据，按二进制位进行“与”运算。
运算规则：
0&0=0 0&1=0 1&0=0 1&1=1
负数按补码形式参加按位与运算

与运算的用途：

1.清零
2.取一个数的指定位
3.判断奇偶

按位或运算(|)

定义：参加运算的两个对象，按二进制位进行“或”运算。
运算规则：
0|0=0 0|1=1 1|0=1 1|1=1
即参加运算的两个对象只要有一个为1，其值为1。

或运算的用途：

1.常用来对一个数据的某些位设置为1

异或运算符（^）

定义：参加运算的两个数据，按二进制位进行“异或”运算。
运算规则：
0^0=0 0^1=1 1^0=1 1^1=0
即参加运算的两个对象，如果两个相应位相同为0，相异为1

异或运算的用途：

1.翻转指定位
2.与0相异或值不变
3.交换两个数

取反运算符（~）

取反运算符 (~)
定义：参加运算的一个数据，按二进制进行“取反”运算。
运算规则：　
~1=0
~0=1

异或运算的用途：

1.使一个数的最低位为零
使a的最低位为0，可以表示为：a & _1。1的值为 1111 1111 1111 1110，再按"与"运算，最低位一定为0。因为“ ~”运算符的优先级比算术运算符、关系运算符、逻辑运算符和其他运算符都高。