POJ 1222 EXTENDED LIGHTS OUT

POJ_1222

    第一次接触高斯消元的题目，基本算是仿写出来的。由于题目没有说无解，也没有用special judge，所以应该按有唯一解的情况去做就可以了。

    当然这个题目也可以用枚举第一行的操作+搜索的办法去解，一些思路可以参考我的POJ_1753的解题报告：http://www.cnblogs.com/staginner/archive/2011/10/29/2228784.html。

#include<stdio.h>

#include<string.h>

#include<iostream>

#define MAXD 35

using namespace std;

int dx[] = {0, -1, 1, 0, 0}, dy[] = {0, 0, 0, -1, 1};

int mat[MAXD][MAXD], ans[MAXD];

void init()

{

    int i, j, k, x, y;

    memset(mat, 0, sizeof(mat));

    for(i = 0; i < 30; i ++)

        scanf("%d", &mat[i][30]);

    for(i = 0; i < 5; i ++)

        for(j = 0; j < 6; j ++)

            for(k = 0; k < 5; k ++)

            {

                x = i + dx[k], y = j + dy[k];

                if(x >= 0 && x < 5 && y >= 0 && y < 6)

                    mat[i * 6 + j][x * 6 + y] = 1;

            }

}

void gauss()

{

    int i, j, k;

    for(i = 0; i < 30; i ++)

    {

        if(mat[i][i] == 0)

        {

            for(j = i + 1; j < 30; j ++)

                if(mat[j][i])

                {

                    for(k = i; k <= 30; k ++)

                        swap(mat[i][k], mat[j][k]);

                }

        }

        for(j = i + 1; j < 30; j ++)

            if(mat[j][i])

            {

                for(k = i; k <= 30; k ++)

                    mat[j][k] ^= mat[i][k];

            }

    }

}

void solve()

{

    int i, j;

    gauss();

    for(i = 29; i >= 0; i --)

    {

        ans[i] = mat[i][30];

        for(j = i + 1; j < 30; j ++)

            ans[i] ^= mat[i][j] * ans[j];

    }

    for(i = 0; i < 5; i ++)

    {

        printf("%d", ans[i * 6]);

        for(j = 1; j < 6; j ++)

            printf(" %d", ans[i * 6 + j]);

        printf("\n");

    }

}

int main()

{

    int t, tt;

    scanf("%d", &t);

    for(tt = 0; tt < t; tt ++)

    {

        init();

        printf("PUZZLE #%d\n", tt + 1);

        solve();

    }

    return 0;

}