矩阵
题目背景
ZSOI2008 T1
题目描述
Neo 有一种很强的计算能力,他能瞬间计算出两个矩阵的乘积,但是他的不足是可能会算错,所以你的任务是对于给出的两个矩阵,检查Neo 计算出的结果是否正确。
注意给出的矩阵都是 N*N 的 01 矩阵,矩阵的计算也是在二进制上的,即:
1*1=1,1*0=0*1=0*0=0,1+1=0+0=0,1+0=0+1=1。
还记得矩阵的计算方法吗?假设 Cn*n = An*n * Bn*n ,那么 Ci,j = ∑kAi,k * Bk,j 。
输入格式
输入包含多组数据,输入第一行为数据组数 T(T≤5)。
对于每组输入数据:第一行为一个整数 N ;接下来共有 3*N 行,每行 N 个数字,取值 0 或 1 ,数据间无空格。数据前面 N 行描述了矩阵 A ,中间 N 行描述了矩阵 B ,最后 N 行为 Neo 计算出来的矩阵 C 。
输出格式
对于每组输入数据,如果Neo的计算是正确的,输出“YES”,否则输出“NO”。
样例数据 1
输入
2
2
10
01
11
10
11
01
3
111
111
000
100
010
001
111
111
000
输出
NO
YES
备注
【数据范围】
对于 50% 的数据,N≤100;
对于 100% 的数据,N≤1000。
解析:
第一次碰到这种玄学随机算法。。。
首先暴力)肯定不行,那么如何应该判断?
玄学的地方来了,我们随机一个一维矩阵,如果满足
,那么答案可能正确,所以多随机几次就有大概率正确。。。
代码:
#include <bits/stdc++.h>
using namespace std;
const int Max=1010;
int n,t,flag;
int a[Max][Max],b[Max][Max],c[Max][Max];
int d[Max],e[Max],f[Max];
char ch[1010];
inline void mul(int a[Max],int b[Max][Max],int c[Max])
{
for(register int j=1;j<=n;j++)
{
c[j]=0;
for(register int k=1;k<=n;k++)
c[j]=(c[j]+a[k]*b[k][j])%2;
}
}
int main()
{
srand(time(0));
scanf("%d",&t);
while(t--)
{
flag=0;
scanf("%d\n",&n);
for(int i=1;i<=n;i++)
{
scanf("%s\n",ch+1);
for(int j=1;j<=n;j++) a[i][j]=ch[j]-'0';
}
for(int i=1;i<=n;i++)
{
scanf("%s\n",ch+1);
for(int j=1;j<=n;j++) b[i][j]=ch[j]-'0';
}
for(int i=1;i<=n;i++)
{
scanf("%s\n",ch+1);
for(int j=1;j<=n;j++) c[i][j]=ch[j]-'0';
}
for(int k=1;k<=10;k++)
{
for(int i=1;i<=n;i++) d[i]=rand()%2;
mul(d,a,e),mul(e,b,f),mul(d,c,e);
for(int i=1;i<=n;i++) if(e[i]!=f[i]) {flag=1;break;}
if(flag) break;
}
if(flag) cout<<"NO\n";
else cout<<"YES\n";
}
return 0;
}