【bzoj2396】神奇的矩阵

Description

给出三个行数和列数均为N的矩阵A、B、C，判断A*B=C是否成立。

Input

题目可能包含若干组数据。
对于每组数据，第一行一个数N，接下来给出三个N*N的矩阵，依次为A、B、C三个矩阵。
Output

对于每组数据，若A*B=C成立，则输出Yes，否则No。每个答案占一行。
Sample Input

1

2

2

100

Sample Output

No

HINT

对于90%的数据，N不超过100；

对于100%的数据，N不超过1000，矩阵中的数字大于等于0小于1000，数据组数不超过5组。

题解
随机生成一个N乘1的矩阵R
然后判断A*B*R是否等于C*R，而前者相当于A*(B*R)
与后者一样都可以在O(N2)的时间里算出来
如果算出来的结果相等
A*B和C几乎也相等。

代码

#include<bits/stdc++.h>
#define N 1005
#define ll long long
#define mo 1000000007
using namespace std;
int rnd[N],a[N][N],b[N][N],c[N][N],t[N],ans1[N],ans2[N];
int n;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
void mul(int a[N][N],int b[N],int c[N])
{
    int ans[N];
    for (int i=1;i<=n;i++)
    {
        ans[i]=0;
        for (int j=1;j<=n;j++) ans[i]+=a[i][j]*b[j];    
    }
    for (int i=1;i<=n;i++) c[i]=ans[i];
}
int main()
{
    for (int i=1;i<=1000;i++)
        rnd[i]=rand();
    while (scanf("%d",&n)!=EOF)
    {
        for (int i=1;i<=n;i++)
            for (int j=1;j<=n;j++)
                a[i][j]=read();
        for (int i=1;i<=n;i++)
            for (int j=1;j<=n;j++)
                b[i][j]=read();
        for (int i=1;i<=n;i++)
            for (int j=1;j<=n;j++)
                c[i][j]=read();
        mul(b,rnd,ans1);
        mul(a,ans1,ans1);
        mul(c,rnd,ans2);
        bool flag=1;
        for (int i=1;i<=n;i++) if (ans1[i]!=ans2[i]){flag=0;break;}
        if (flag) puts("Yes");else puts("No");
    }
    return 0;
}