POJ 1222 (枚举+熄灯问题)

http://poj.org/problem?id=1222

题意：5x6个灯输出怎么熄灯才能都是灭的。0是灭，1是亮。

思路：首先要确定第一行的灯，随后剩下的灯随之确定，因为如果puzzle[r][c]是亮的，则要按下press[2][j]。

怎么确定第一行的灯呢，第一行有2^6种情况，依次枚举即可。

要点有三处：

   1：对puzzle和press的初始化。很重要，也是为了第三步的计算。

   2：枚举2^6的方法很奇妙，可以自己演算一下。

   3：press[r+1][c]的值可以通过press[r][c]四周的状态来计算，别忘了加puezzle[r][c]这个初始状态。

#include<cstdio>

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

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

void enumate()
{
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]++;
}
}
}

int main()
{
freopen("in.txt","r",stdin);
int ncase,r,c;
scanf("%d",&ncase);

for(r = 1;r < 6; r++)
press[r][0] = press[r][7] = 0;
for(c = 1;c < 7; c++)
press[0][c] = 0;
for(int i = 1;i <= ncase; i++){

for(r = 1;r <= 5; r++)
for(c = 1;c <= 6; c++)
scanf("%d",&puzzle[r][c]);

enumate();
printf("PUZZLE #%d\n",i);
for(r = 1;r <= 5; r++){

for(c = 1;c < 7; c++)
printf("%d ",press[r][c]);

printf("\n");
}
}
return 0;
}