EXTENDED LIGHTS OUT POJ - 1222

EXTENDED LIGHTS OUT POJ - 1222

题意:

给一个T,有T组数据,每组数据有5行6列的0-1数据,0代表关,1代表开,现在要求设计开关的状态,来把数据中所有的1都变成0;

分析:

只要第一行确定了,那么下面每一行都是确定的,因为第二行肯定是为了让第一行所有的为0,以此类推,所有到了最后一行,如果最后一行也全部为0,那么开关的状态就是可行的,这里为了方便起见,我们用位运算.

AC代码:

#include <iostream>
#include <string>
#include <cstring>
#include <memory>
using namespace std;

char inputlight[5];
char nowlight[5];
char result[5];

//得到某个字符的第i位
int Getbit(char c,int i){
	return (c>>i) & 1;
}

//将 c 的第 i 位变成 v 
void Setbit(char &c,int i,int v){
	//如果v是1 
	if(v)
		c = c | (1<<i);
	//如果v是0 
	else
		c = c & ~(1<<i); 
}

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

int main(){
	int T;
	cin>>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(inputlight[i],j,s);
			}
		}
		
		
		for(int n=0;n<64;n++){
			int switchs=n;//第一行开关的状态 
			memcpy(nowlight,inputlight,sizeof(inputlight));
			
			for(int i=0;i<5;i++){
				result[i]=switchs;
				for(int j=0;j<6;j++){
					if(Getbit(switchs,j)){//如果开关某个bit的状态是1,需要翻转这一行 
						if(j>0)	
							Flipbit(nowlight[i],j-1);
						if(j<5)	
							Flipbit(nowlight[i],j+1);
						Flipbit(nowlight[i],j);
					} 
				}
				//继续翻转下一行
				if(i<4)
					nowlight[i+1] ^= switchs;
					//这一行还有很多没有灭掉的灯 所以下一行的开关的状态 要和这一行一致 才能灭掉这些灯 
				switchs=nowlight[i];
			}
			
			//当最后一行也为0时  结束 
			if(nowlight[4]==0){
				cout<<"PUZZLE #"<<t<<endl;
				for(int i=0;i<5;i++){
					for(int j=0;j<6;j++)
						cout<<Getbit(result[i],j)<<" ";
					cout<<endl;
				}	
				break;
			}
		}	
	}
	return 0;
}