题目明说不能用一般O(n^3)方法,所以要寻求出路!
首先,我们要知道矩阵连乘运算公式x*a*b=x*c,这样就好办,令x为1 x n矩阵,左边算出未一阶,右边也是一阶,把复杂度降为O(n^2),对于x我们可以用随机函数赋值,也可以不用,直接令x[i]=i。
注意随机函数用法srand((unsigned int)time(0))——种子只要生成一次!记得加头文件<time.h>!
#include <vector>
#include <list>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <deque>
#include <stack>
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <limits.h>
#include <time.h>
using namespace std;
#define LL long long
int lowbit(int t){return t&(-t);}
int countbit(int t){return (t==0)?0:(1+countbit(t&(t-1)));}
LL gcd(LL a,LL b){return (b==0)?a:gcd(b,a%b);}
#define PI acos(-1.0)
#define N 501
#define MAX INT_MAX
#define MIN INT_MIN
#define eps 1e-8
#define FRE freopen("a.txt","r",stdin)
//O(n^2)
LL a[N][N],b[N][N],c[N][N],x1[N],x2[N],x3[N],x4[N];
int main(){FRE;
int n;
int i,j,k;
scanf("%d",&n);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
scanf("%I64d",&a[i][j]);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
scanf("%I64d",&b[i][j]);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
scanf("%I64d",&c[i][j]);
srand((unsigned int)time(0)); //种子只要生成一次!
for(i=1;i<=n;i++)x4[i]=(rand()%10000+0);
for(i=1;i<=n;i++) //O(n^2),原来复杂度是看最大的那个!
for(j=1;j<=n;j++)
{
x1[i]+=a[j][i]*x4[j];
x2[i]+=c[j][i]*x4[j];
}
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
x3[i]+=b[j][i]*x1[j];
for(i=1;i<=n;i++)
if(x3[i]!=x2[i])
break;
if(i<=n)puts("NO");
else
puts("YES");
return 0;
}