poj 1222 经典枚举

点击打开链接

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int M=8;
int g[M][M],button[M][M];
bool solve()
{
	int i,j,k;
	for(i=1;i<5;i++) //根据第i行的state 确定第i+1行的state 
	{
		for(j=1;j<=6;j++)
		{
			button[i+1][j]=(g[i][j]+button[i][j]+button[i][j-1]+button[i][j+1]+button[i-1][j])%2;
		}
	}
	for(j=1;j<=6;j++)
	{
		//原来如果1   总的按钮产生的效果也要为1  原来0 总的按钮0 
		if(g[5][j]!=(button[5][j]+button[5][j-1]+button[5][j+1]+button[4][j])%2)	//判断最后一行灯是否全灭 
		{
			return false;
		}
	} 
	return true;
}
void enumerate()
{
	int i,j,k;
	k=0;
	for(k=1;k<=6;k++)
	{
		button[1][k]=0;
	}
	while(!solve())
	{
		k=1;
		button[1][k]++;
		while(button[1][k]>1)
		{
			button[1][k]=0;
			k++;
			button[1][k]++;	
		}	
	}

	
}
int main()
{
	int cas,c;
	cin>>cas;
	c=0;
	while(cas--)
	{
		
		for(int i=0;i<=6;i++)
		{
			for(int j=0;j<=7;j++)
			{
				if(i==0||j==0||j==7||i==6)
				{
					g[i][j]=0;
				}
				else
				cin>>g[i][j];
			}
		}
		enumerate();//枚举第一行按钮的state（第一行按完后 第一行中可能有些灯没灭） 为了使全部的灯熄灭 第2行的状态就固定了 （第二行要是第一行的灯全熄灭） 
					//	 依次类推 最后一行使倒数第二行全灭 只要最后一行也熄灭了 则全部的灯都熄灭了 
		printf("PUZZLE #%d\n",++c);
		
		for(int i=1;i<=5;i++)
		{
			for(int j=1;j<=6;j++)
			{
				cout<<button[i][j]<<' ';
			}
			cout<<endl;
		}
	}
	return 0;
}