poj_1222_高斯消元

第一次学习使用高斯消元，将灯板化为线性方程组，进行求解。

/*#########################################################################
# File Name: poj_1222.cpp
# Author: CaoLei
# Created Time: 2015/7/20 15:48:04
#########################################################################*/

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <set>
#include <queue>
#include <map>
using namespace std;
#define MAX(x,y) (((x)>(y))?(x):(y))
#define MIN(x,y) (((x)<(y))?(x):(y))
#define N 500010
#define pi acos(-1.0)
#define inf 100000000
typedef long long ll;
typedef unsigned long long ull;
typedef vector<double> vec;
typedef vector<vec> mat;
const double eps=1e-8;
int a[35][35];
int ans[35];
void gauss(){
    for(int i=0;i<30;i++){
        int tmp=0,k=0;
        for(int j=0;j<30;j++){
            if(a[j][i]>tmp) k=j;
        }
        for(int j=0;j<=30;j++){
            swap(a[i][j],a[k][j]);
        }
        for(int j=0;j<30;j++){
            if(i!=j&&a[j][i])
              for(int k=i+1;k<=30;k++){
                  a[j][k]=a[j][k]^a[i][k];
              }
        }
    }
}

int main(){
    freopen("in.txt","r",stdin);
    int t;
    scanf("%d",&t);
    for(int cnt=1;cnt<=t;cnt++){
        printf("PUZZLE #%d\n",cnt);
        memset(a,0,sizeof(a));
        for(int i=0;i<30;i++){
            scanf("%d",&a[i][30]);
            ans[i]=0;
        }
        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();
        for(int i=0;i<30;i++){
            ans[i]=a[i][30];
        }
        for(int i=0;i<30;i++){
            if((i+1)%6==0)
                printf("%d\n",ans[i]);
            else printf("%d ",ans[i]);
        }
    }
    return 0;
}

版权声明：本文为博主原创文章，未经博主允许不得转载。