POJ 1830 开关问题 (高斯消元)

题目地址:POJ 1830

高斯消元第一发。一个地方逻辑判断出现了失误,调了一下午啊。。。

通过高斯消元来找矩阵的秩,然后2^(自由元的数量)就是答案。因为对于每个自由元,都有0和1两种状态可选。

代码如下:

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
using namespace std;
#define LL __int64
#define pi acos(-1.0)
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double eqs=1e-6;
int a[35], b[35], d[35][35], n;
int gauss()
{
        int i, j, k, r, h;
        for(i=0,j=0; i<n&&j<n;i++,j++) {
                r=i;
                if(d[i][j]==0) {
                        for(k=i+1; k<n; k++) {
                                if(d[k][j]) {
                                        r=k;
                                        break;
                                }
                        }
                        if(k>=n){
                                i--;continue ;
                        }
                }
                if(r!=i) {
                        for(k=0; k<=n; k++) {
                                swap(d[i][k],d[r][k]);
                        }
                }
                for(k=i+1; k<n; k++) {
                        if(d[k][j]==0) continue ;
                        for(h=j; h<=n; h++) {
                                d[k][h]^=d[i][h];
                        }
                }
        }
        for(k=i; k<n; k++) {
                if(d[k][n]) return -1;
        }
        return 1<<n-i;
}
int main()
{
        int t, i, x, y, ans;
        scanf("%d",&t);
        while(t--) {
                scanf("%d",&n);
                memset(d,0,sizeof(d));
                for(i=0; i<n; i++) {
                        scanf("%d",&a[i]);
                }
                for(i=0; i<n; i++) {
                        scanf("%d",&b[i]);
                        if(a[i]!=b[i]) {
                                d[i][n]=1;
                        }
                        d[i][i]=1;
                }
                while(scanf("%d%d",&x,&y)!=EOF&&x+y) {
                        d[y-1][x-1]=1;
                }
                ans=gauss();
                if(ans<0) {
                        printf("Oh,it's impossible~!!\n");
                } else {
                        printf("%d\n",ans);
                }
        }
        return 0;
}


你可能感兴趣的:(编程,算法,C语言,ACM,高斯消元)