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;
}