【随机算法】poj 3318

题目明说不能用一般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;
}