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个全部关联的情形有
2n−1种
情形,所以只要每一种都需要枚举到,一定会超时。
那么本题只能用异或方程组解,用高斯消元法解决。需要用到高等代数知识。出现0=1则无解,否则若不定元有k个,答案为
2k种
。
本题还有一种想法就是设系数矩阵为A,增广矩阵为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))