URAL 1042 Central Heating

URAL_1042

    由于题目中说明了每个工人不能被其他工人取代，也就是说如果矩阵的列向量是线性无关的，于是增广矩阵的秩一定是N，所以不会有无解的情况，并且解是唯一的，所以直接用高斯消元求解即可。

#include<stdio.h>
#include<string.h>
#include<iostream>
#define MAXD 260
using namespace std;
int N, mat[MAXD][MAXD], ans[MAXD];
void init()
{
    int i, j, k;
    memset(mat, 0, sizeof(mat));
    for(i = 0; i < N; i ++)
    {
        for(;;)
        {
            scanf("%d", &j);
            if(j == -1)
                break;
            mat[j - 1][i] = 1;
        }
        mat[i][N] = 1;
    }
}
void gauss()
{
    int i, j, k;
    for(i = 0; i < N; i ++)
    {
        if(mat[i][i] == 0)
        {
            for(j = i + 1; j < N; j ++)
                if(mat[j][i])
                {
                    for(k = i; k <= N; k ++)
                        swap(mat[i][k], mat[j][k]);
                    break;
                }
        }
        for(j = i + 1; j < N; j ++)
            if(mat[j][i])
            {
                for(k = i; k <= N; k ++)
                    mat[j][k] ^= mat[i][k];
            }
    }
    for(i = N - 1; i >= 0; i --)
    {
        ans[i] = mat[i][N];
        for(j = i + 1; j < N; j ++)
            ans[i] ^= mat[i][j] * ans[j];
    }
}
void solve()
{
    int i, j, k, flag = 0;
    gauss();
    for(i = 0; i < N; i ++)
        if(ans[i])
        {
            if(flag)
                printf(" ");
            else
                flag = 1;
            printf("%d", i + 1);
        }
    printf("\n");
}
int main()
{
    while(scanf("%d", &N) == 1)
    {
        init();
        solve();
    }
    return 0;
}