poj 1222 EXTENDED LIGHTS OUT 高斯消元

传送门

题目就是给你一些灯， 点击一个灯， 就会改变它自身和它四周的所有的灯的状态。 给你所有灯的初始状态， 要你输出所有灯的点击情况， 1代表点击， 0代表不点击。

#include<bits/stdc++.h>
using namespace std;
#define mem(a) memset(a, 0, sizeof(a))
const int dir[][2] = { {-1, 0}, {1, 0}, {0, -1}, {0, 1} };
int a[35][35], ans[35];
void gauss() {
    for(int k = 0; k<30; k++) {
        int i = k;
        while(i<30&&!a[i][k])
            i++;
        if(i>k) {
            for(int r = 0; r<=30; r++)
                swap(a[i][r], a[k][r]);
        }
        for(int i = 0; i<30; i++) {
            if(i!=k && a[i][k]) {
                for(int j = 0; j<=30; j++) {
                    a[i][j] ^= a[k][j];
                }
            }
        }
    }
    for(int i = 0; i<30; i++) {
        if(a[i][30]) {
            int j = 0;
            while(j<30&&!a[i][j])
                j++;
            ans[j] = a[i][30];
        }
    }
}
int main()
{
    int t;
    cin>>t;
    for(int cnt = 1; cnt<=t; cnt++) {
        mem(a);
        mem(ans);
        for(int i = 0; i<30; i++) {
            scanf("%d", &a[i][30]);
        }
        for(int i = 0; i<30; i++)
            a[i][i] = 1;
        for(int i = 0; i<5; i++) {
            for(int j = 0; j<6; j++) {
                for(int k = 0; k<4; k++) {
                    int x = i+dir[k][0];
                    int y = j+dir[k][1];
                    if(x>=0&&x<5&&y>=0&&y<6) {
                        a[i*6+j][x*6+y] = 1;
                    }
                }
            }
        }
        gauss();
        printf("PUZZLE #%d
", cnt);
        for(int i = 0; i<30; i++) {
            int x = i/6;
            int y = i%6;
            a[x][y] = ans[i];
        }
        for(int i = 0; i<5; i++) {
            printf("%d", a[i][0]);
            for(int j = 1; j<6; j++) {
                printf(" %d", a[i][j]);
            }
            puts("");
        }
    }
}