BZOJ2396 随机化

直接乘n3肯定GG 考虑建立一个新的1*n的矩阵R A*B*R=C*R R随机化 最后几乎一样就是成立的

#include<bits/stdc++.h>
using namespace std;
int n;
int a[1001][1001],b[1001][1001],c[1001][1001];
int rnd[1001],ans1[1001],ans2[1001]; 
void mmul(int a[1001],int b[1001][1001],int s[1001])
{
    int tmp[1001];
    for(int j=1;j<=n;j++)
    {
        tmp[j]=0;
        for(int k=1;k<=n;k++)
           tmp[j]+=a[k]*b[k][j];
    }
    for(int i=1;i<=n;i++)s[i]=tmp[i];
}
bool judge()
{
    for(int i=1;i<=n;i++)
        if(ans1[i]!=ans2[i])return 0;
    return 1;
}
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++)
                scanf("%d",&a[i][j]);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&b[i][j]);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&c[i][j]);
        mmul(rnd,a,ans1);
        mmul(ans1,b,ans1);
        mmul(rnd,c,ans2);
        if(judge())printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}