poj 1830 开关问题(异或方程组+高斯消元法)

poj 1830 开关问题(异或方程组+高斯消元法)
Time Limit: 1000ms Memory Limit: 65536kB

Description
有N个相同的开关，每个开关都与某些开关有着联系，每当你打开或者关闭某个开关的时候，其他的与此开关相关联的开关也会相应地发生变化，即这些相联系的开关的状态如果原来为开就变为关，如果为关就变为开。你的目标是经过若干次开关操作后使得最后N个开关达到一个特定的状态。对于任意一个开关，最多只能进行一次开关操作。你的任务是，计算有多少种可以达到指定状态的方法。（不计开关操作的顺序）

Input
输入第一行有一个数K，表示以下有K组测试数据。
每组测试数据的格式如下：
第一行 一个数N（0 < N < 29）
第二行 N个0或者1的数，表示开始时N个开关状态。
第三行 N个0或者1的数，表示操作结束后N个开关的状态。
接下来 每行两个数I J，表示如果操作第 I 个开关，第J个开关的状态也会变化。每组数据以 0 0 结束。

Output
如果有可行方法，输出总数，否则输出“Oh,it’s impossible~!!” 不包括引号

Sample Input

2
3
0 0 0
1 1 1
1 2
1 3
2 1
2 3
3 1
3 2
0 0
3
0 0 0
1 0 1
1 2
2 1
0 0

Sample Output

4
Oh,it’s impossible~!!

Hint
第一组数据的说明：
一共以下四种方法：
操作开关1
操作开关2
操作开关3
操作开关1、2、3 （不记顺序）

Source
LIANGLIANG@POJ

---

首先本题用枚举的话，无论如何剪枝，由于n个全部关联的情形有$2^{n-1}种$情形，所以只要每一种都需要枚举到，一定会超时。
那么本题只能用异或方程组解，用高斯消元法解决。需要用到高等代数知识。出现0=1则无解，否则若不定元有k个，答案为$2^k种$。
本题还有一种想法就是设系数矩阵为A，增广矩阵为A*，则

$$解的数目=2^{n-rank(A)}\ \ \ \ \ 当rank(A)=rank(A*)时，否则无解$$
由于python的numpy库含现成的rank，不必要再高斯消元法实现了(但是poj不支持python的numpy库)。

---

Accepted    128kB   1ms 1526 B  G++

C++ version

#include<stdio.h>

const int N=30;
const char* FAIL="Oh,it's impossible~!!\n";

bool r[N][N+1];

void Gauss(bool r[N][N+1],int n)
{
    int i0=-1,j0,temp;
    for (int i=0;i<n;i++)
    {
        j0=-1;
        for (int j=i;j<n;j++)
            if (r[j][i])
            {
                j0=j;
                break;
            }
        if (j0!=-1)
        {
            i0++;
            if (j0!=i0)
            {
                for (int j=i;j<n+1;j++)
                {
                    temp=r[i0][j];
                    r[i0][j]=r[j0][j];
                    r[j0][j]=temp;
                }
            }   
            for (int j=i0+1;j<n;j++)
                if (r[j][i])
                    for (int k=i;k<n+1;k++)
                        r[j][k]^=r[i0][k];
        }
    } 
    return;
}

void print(bool r[N][N+1],int n)
{
    for (int i=0;i<n;i++)
        for (int j=0;j<n+1;j++)
            printf("%d%c",r[i][j],j==n?'\n':' ');
    printf("\n");
    return; 
}

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    int cases,n,x,y,rank1,rank2;
    bool zero;
    scanf("%d",&cases);
    while (cases--)
    {
        scanf("%d",&n);
        for (int i=0;i<n;i++)
            for (int j=0;j<n+1;j++)
                r[i][j]=false;
        for (int i=0;i<n;i++)
            r[i][i]=true;
        for (int i=0;i<n;i++)
            scanf("%d",&r[i][n]);
        for (int i=0;i<n;i++)
        {
            scanf("%d",&x);
            r[i][n]^=x;
        }
        while (scanf("%d%d",&x,&y)&&x&&y)
            r[y-1][x-1]=true;
        //print(r,n);
        Gauss(r,n);
        //print(r,n);
        rank1=n;
        for (int i=0;i<n;i++)
        {
            zero=true;
            for (int j=0;j<n;j++)
                if (r[i][j])
                    zero=false;
            if (zero)
            {
                rank1=i;
                break;
            }
        }
        rank2=rank1;
        for (int i=rank1;i<n;i++)
            if (r[i][n])
                rank2++;
        if (rank1==rank2)
            printf("%d\n",1<<(n-rank1));
        else
            printf(FAIL); 
    }
    return 0;
}

python3 version

from numpy import *

cases=int(input())
for case_no in range(cases):
    n=int(input())
    A=zeros([n,n+1],bool)
    for i in range(n):
        A[i][i]=1;

    line=input()
    for i in range(n):
        if (i!=n-1):
            now,line=line.split(' ',1)
        else:
            now=line
        A[i][n]=int(now)
    line=input()
    for i in range(n):
        if (i!=n-1):
            now,line=line.split(' ',1)
        else:
            now=line
        A[i][n]=(A[i][n]+int(now))%2

    while (True):
        x,y=input().split()
        x,y=int(x)-1,int(y)-1
        if ((x==-1) and (y==-1)):
            break
        A[y][x]=1

    rank1=linalg.matrix_rank(A[0:n,0:n])
    rank2=linalg.matrix_rank(A)
    if (rank1!=rank2):
        print("Oh,it\'s impossible~!!")
    else:
        print(2**(n-rank1))