POJ 1222

题目POJ 1222

新手一枚，还没学算法，用的枚举法。

解题思路：枚举操作的第一行，从第一行开始依次往下操作，如此当第i行操作结束，满足让第i行灯全灭的操作条件只能是optLight[i+1]=resLight[i]，判断最后一行是否全灭。

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

void Pri(int a[5][6]);
 
int main()
{
	int oriLight[5][6];//初始 
	int optLight[5][6];//操作 
	int resLight[5][6];//结束
	memset(optLight,0,sizeof(optLight));
	int T,flag=1;
	cin>>T;
	for(int n=1;n<=T;n++){
		for(int i=0;i<5;i++)
			for(int j=0;j<6;j++)
				cin>>oriLight[i][j];//输入 
		for(optLight[0][0]=0;optLight[0][0]<2;optLight[0][0]++)
		for(optLight[0][1]=0;optLight[0][1]<2;optLight[0][1]++)
		for(optLight[0][2]=0;optLight[0][2]<2;optLight[0][2]++)
		for(optLight[0][3]=0;optLight[0][3]<2;optLight[0][3]++)
		for(optLight[0][4]=0;optLight[0][4]<2;optLight[0][4]++)
		for(optLight[0][5]=0;optLight[0][5]<2;optLight[0][5]++){//枚举opt第一行 
		memcpy(resLight,oriLight,sizeof(oriLight));
		for(int i=0;i<5;i++){//操作
			for(int j=0;j<6;j++){ 
				if(optLight[i][j]){
					resLight[i][j]++;
					if(j>0) resLight[i][j-1]++;
					if(j<5) resLight[i][j+1]++;
					if(i<4) 
					resLight[i+1][j]++;
					if(i>0)
					resLight[i-1][j]++;
				}
			}
			if(i<4){
				for(int j=0;j<6;j++)	
					optLight[i+1][j]=resLight[i][j]%2;
			}			
		}
		flag=1;
		for(int j=0;j<6;j++) 
			if(resLight[4][j]%2){
				flag=0;
				break;
			}
		if(flag){
			cout<<"PUZZLE #"<<n<<endl;
			Pri(optLight);
			break;
		}
		}
	}
	return 0;
 } 
 
 void Pri(int a[5][6]){
 	for(int i=0;i<5;i++){
		for(int j=0;j<6;j++){
			cout<<a[i][j]%2;
			if(j<5) cout<<" ";
		}
		cout<<endl;;
	}
 }