poj1222 EXTENDED LIGHTS OUT

设输入矩阵为A，输出矩阵为B，目标矩阵为C（零矩阵）。

方便起见，矩阵行列下标均从1开始。

考虑A矩阵元素a（i，j），B矩阵中与其相邻的元素 b（i，j），b（i - 1， j），b（i + 1，j），b（i，j - 1），b（i，j + 1） （#）。

有c（i，j）= 0 = a（i，j） ^ （∑ b（i，j） % 2）。

进一步有a（i，j） ^ 0 = a（i，j） = a（i，j） ^ a（i，j） ^ （∑ b（i，j） % 2） = ∑ b（i，j） % 2     (*)。

进一步考虑异或与模2加之间的关系：∑ b（i，j） % 2 = ^ B(i, j)，其中B（i，j）表示（#）式5个元素的集合。

带入（*）：^ B(i, j) = a（i，j）。这是我们需要的方程组。

（注：考虑边界上的元素，只需将不在矩阵范围内的b元素全部置零即可。）

将b（i，j）映射到x（（i - 1） * 5 + j）（x下标从1到5 * 6)。

下面考虑解异或方程组AX= B。（与前面的表达无关）

a11 * x1 ^ a12 * x2 ^...^ a1n * xn = b1 ①

...

an1 * x1 ^ an2 * x2 ^...^ann * xn = bn ②

A,X,B均为0-1矩阵。

考虑消元，现在看系数矩阵A的第一列，若全部元素均为0，直接转到下一列，并且有方程组有多解。

若存在ai1 = 1,将其与第一行交换（这样做的目的是为了得到上三角阵）。

此时只需考虑剩余所有 1 < j ≤ n, 且aj1 = 1。有：

a11 * x1 ^ a12 * x2 ^...^ a1n * xn = b1 ①

aj1 * x1 ^ aj2 * x2 ^...^ ajn * xn = bj ②

① ^ ②：

（a11 * x1 ^ a12 * x2 ^...^ a1n * xn） ^ （aj1 * x1 ^ aj2 * x2 ^...^ ajn * xn） = b1 ^ bj。

即：（a11 * x1 ^ aj1 * x1） ^ （a12 * x2 ^ aj2 * x2） ^...^ （a1n * xn ^ ajn * xn) = b1 ^ bj 。

易于验证：a * x ^ b ^ x = (a ^ b) * x，因此进一步有：

（a11 ^ aj1）*x1 ^ （a12 ^ aj2）*x2  ^ .... ^ （a1n ^ ajn）*xn = b1 ^ bj 。

因而第j行x1的系数更新为1 ^ 1 即0 。 

对增广阵如此消元至得到上三角阵即可得到答案。

http://poj.org/problem?id=1222

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxx = 5;
const int maxy = 6;
const int maxm = maxx * maxy;
const int maxn = maxm + 10;
int A[maxn][maxn], B[maxn];
int C[maxn][maxn];
int dx[] = {0, 0, 0, -1, 1};
int dy[] = {0, -1, 1, 0, 0};
int pos(int i, int j) { return (i - 1) * maxy + j; }
bool in_range(int i, int j) { return i >= 1 && i <= maxx && j >= 1 && j <= maxy; }
void solve(){
    memset(C, 0, sizeof C);
    for(int i = 1, k = 1; i <= maxx; i++) for(int j = 1; j <= maxy; j++, k++){
        //the k-th column of the coefficient matrix
        for(int u = 0; u < 5; u++){
            int nx = i + dx[u], ny = j + dy[u], p = pos(nx, ny);
            if(in_range(nx, ny)) C[k][p] = 1;
        }
        C[k][maxm + 1] = A[i][j];
    }
    for(int i = 1; i <= maxm; i++){
        //eliminate for X(i)
        for(int j = i; j <= maxm; j++){
            //highlighting A[j][]
            if(C[j][i]){
                swap(C[i], C[j]);
                break;
            }
        }
        for(int j = i + 1; j <= maxm; j++){
            if(!C[j][i]) continue;
            for(int k = i; k <= maxm + 1; k++) C[j][k] ^= C[i][k];
        }
    }
    //retrieve the ans
    for(int i = maxm; i >= 1; i--){
        B[i] = C[i][maxm + 1];
        for(int j = i + 1; j <= maxm; j++) if(C[i][j]) B[i] ^= B[j];
    }
    for(int i = 1; i <= maxx; i++){
        printf("%d", B[pos(i, 1)]);
        for(int j = 2; j <= maxy; j++){
            int p = pos(i, j);
            printf(" %d", B[p]);
        }
        printf("
");
    }
}

int main(){
    freopen("in.txt", "r", stdin);
    int T, kase = 0;
    scanf("%d", &T);
    while(T--){
        printf("PUZZLE #%d
", ++kase);
        for(int i = 1; i <= maxx; i++) for(int j = 1; j <= maxy; j++) scanf("%d", &A[i][j]);
        solve();
    }
    return 0;
}