poj 1222 EXTENDED LIGHTS OUT（数学：高斯消元||爆搜：DFS）

因为上一行的状态仅与当前行有关

所以我们可以考虑枚举第一行的1<<6个状态

之后再从第2行到第6行开始根据上一行的状态而改变

时间复杂度是O（(1<<6)*5*6）

代码如下：

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

int a[10][10];
int b[10][10];
int c[10][10];

void change(int x, int y) {
    a[x][y] = 1-a[x][y];
    a[x-1][y] = 1-a[x-1][y];
    a[x+1][y] = 1-a[x+1][y];
    a[x][y-1] = 1-a[x][y-1];
    a[x][y+1] = 1-a[x][y+1];
    return ;
}

void dfs(int x) {
    for(int i=0; i<6; ++i) {
        if(x & (1<<i)) {
            b[1][i+1] = 1;
            change(1, i+1);
        }
    }
    return ;
}

void print() {
    for(int i=1; i<=5; ++i) {
        printf("%d", b[i][1]);
        for(int j=2; j<=6; ++j) 
            printf(" %d", b[i][j]);
        puts("");
    }
    return ;
}

bool solve() {

    for(int i=2; i<=5; ++i) {
        for(int j=1; j<=6; ++j) {
            if(a[i-1][j] == 1) {
                b[i][j] = 1;
                change(i, j);
            }
        }
    }
    for(int i=1; i<=5; ++i) {
        for(int j=1; j<=6; ++j)
            if(a[i][j])
                return false;
    }
    return true;
}

int main(void) {
    int n;
    scanf("%d", &n);
    for(int _=1; _<=n; ++_) {
        memset(a, 0, sizeof(a));
        for(int i=1; i<=5; ++i) 
            for(int j=1; j<=6; ++j) 
                scanf("%d", &c[i][j]);
        for(int i=0; i<(1<<6); ++i) {
            memcpy(a, c, sizeof(c));
            memset(b, 0, sizeof(b));
            dfs(i);
            /*
            puts("");
            puts("");
            puts("");
            */
            if(solve()) {
             //   printf("b--->\n");
                printf("PUZZLE #%d\n", _);
                print();
                break;
            }
            /*
            printf("c--->\n");
            for(int i=1; i<=5; ++i) {
                for(int j=1; j<=6; ++j) {
                    printf("%d", c[i][j]);
                }
                puts("");
            }
            puts("");
            puts("");
            puts("");
            */
        }
    }
    return 0;
}

高斯消元法的做法是列出30个方程，并由这些方程求出30个未知数

而这里构造方程的方法很值得学习以下

因为一个方程中可能含3、4、5个未知数

对应一个点周围的点

由此可以构造一个30×30的矩阵

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

int a[100][100];
void gauss() {
    int i, j, k, r;
    for(i=0; i<30; ++i) {
        r = i;
        for(j=i+1; j<30; ++j)
            if(a[j][i]) 
                r = j;
        if(r != i) {
            for(k=0; k<=30; ++k)
                swap(a[i][k], a[r][k]);
        }

        for(k=0; k<30; ++k) {
            if(k!=i && a[k][i])
                for(j=0; j<=30; ++j)
                    a[k][j] ^= a[i][j];
        }
    }
    return ;
}

int main(void) {
    int n;
    scanf("%d", &n);
    for(int _=1; _<=n; ++_) {
        for(int i=0; i<30; ++i) {
            scanf("%d", &a[i][30]);
        }
        for(int i=0; i<30; ++i) {
            a[i][i] = 1;
            if(i%6 != 0) a[i-1][i] = 1;
            if(i%6 != 5) a[i+1][i] = 1;
            if(i > 5) a[i-6][i] = 1;
            if(i < 24) a[i+6][i] = 1;
        }
        
        gauss();
        printf("PUZZLE #%d\n", _);
        for(int i=0; i<30; ++i) {
            printf("%d", a[i][30]);
            if(i%6 == 5) puts("");
            else printf(" ");
        }
    }
    return 0;
}