noiopenjudge1813:熄燈問題 貪心

noiopenjudge1813:熄燈問題

貪心：
題解寫在題面上。

可以想到 在處理 i 行 的時候保證 i 行之前的狀態一定是合法的狀態， i 行之後的不管、
這樣我一遍處理下去。
因爲第一行無論是什麼狀態我都可以通過第二行而改變 所以枚舉一遍第一行的狀態然後處理即可。

#include <iostream>
#include <queue>
#include <cstdio>
#include <cstring>
using namespace std;

const int MAXN = 10;
bool maps[MAXN][MAXN], a[MAXN][MAXN];

bool solve()
{
    for(int i = 1; i <= 5; i ++)
        for(int j = 1; j <= 6; j ++)
            a[i+1][j] = (maps[i][j] + a[i][j] + a[i][j-1] + a[i][j+1] + a[i-1][j]) % 2;

    for(int j = 1; j <= 6; j ++)
        if((maps[5][j] + a[5][j] + a[5][j-1] + a[5][j+1] + a[5-1][j])%2)
            return 0;

    for(int i = 1; i <= 5; i ++)
    {
        for(int j = 1; j <= 6; j ++)
            cout << a[i][j] << " ";
        cout << endl;
    }
    return 1;
}

int main()
{
    for(int i = 1; i <= 5; i ++)
        for(int j = 1; j <= 6; j ++)
            scanf("%d", &maps[i][j]);
    for(int i = 0; i < (1<<6); i ++)
    {
        memset(a,0,sizeof(a));
        for(int j = 0; j < 6; j ++)
        {
            if(i&(1<<j)) // 如果是 1;
                a[1][6-j] = 1;
        }
        if(solve())
            return 0;
    }
    return 0;
}