POJ 1222

熄燈問題

幾個TIPS

1、只要有第一行的初始狀態和第一行是否按燈的情況就能確定下面所有行的按燈情況。因爲上面的燈亮着，對應位置的下面的開關就必須點一下，反之則不能點。

2、將5*6 的矩陣擴充爲6*8的矩陣，使每個開關的情況變成一樣。

3、6重循環用僞二進制進位的方法看起來更符合邏輯一些。

#include<iostream>
#include <stdio.h>
using namespace std;

int puzzle[6][8],press[6][8];

bool guess()
{
	int c,r;
	for (r=1;r<5;r++)
	{
		for (c=1;c<7;c++)
		{
			press[r+1][c]=(puzzle[r][c]+press[r][c-1]+press[r][c]+press[r][c+1]+press[r-1][c])%2;
		}
	}
	for(c=1;c<7;c++)
	{
		if((press[5][c-1]+press[5][c]+press[5][c+1]+press[4][c])%2!=puzzle[5][c])
			return false;
	}
	return true;
}

void enumerate()
{
	int c;
	for(c=1;c<7;c++)
		press[1][c]=0;
	while(guess()==false)
	{
		press[1][1]++;
		c=1;
		while(press[1][c]>1)
		{
			press[1][c]=0;
			c++;
			press[1][c]++;
		}
	}
	return;
}

int main()
{
	int cases,i,r,c;
	scanf("%d",&cases);
	for(r=0;r<6;r++)
	{
		press[r][0]=press[r][7]=0;
	}
	for(c=1;c<7;c++)
	{
		press[0][c]=0;
	}
	for(i=0;i<cases;i++)
	{
		for(r=1;r<6;r++)
			for(c=1;c<7;c++)
				scanf("%d",&puzzle[r][c]);
		enumerate();
		cout<<"PUZZLE #"<<i+1<<endl;
		for(r=1;r<6;r++){
			for(c=1;c<7;c++)
				cout<<press[r][c]<<" ";
			cout<<endl;
		}
	}
	return 0;
}