poj 3318

题目：http://poj.org/problem?id=3318

题目中明确交代，0（n ^ 3) 会 TLE，可是一开始还就没看见，以为在计算矩阵时加两个优化可在 2s 挤过去呢，然后就TLE了三次，看别人的说是有一个结论 A * B * X == C * X时，有 A * B == C成立（成立的概率很大），然后 X 就是用随机函数求的

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <vector>
#include <math.h>
#define N 510
#define mod 10000
#define _clr(a,val) (memset(a,val,sizeof(a)))

using namespace std;

typedef long long ll;

int a[N][N],b[N][N],c[N][N];
int ta[N],tb[N],tc[N];
int n;
int main()
{
    int i,j;
    //freopen("data.txt","r",stdin);
    while(cin>>n)
    {
        for(i = 1; i <= n; i++)
        {
            for(j = 1; j <= n; j++)
            scanf("%d",&a[i][j]);
        }
        for(i = 1; i <= n; i++)
        {
            for(j = 1; j <= n; j++)
            scanf("%d",&b[i][j]);
        }
        for(i = 1; i <= n; i++)
        {
            for(j = 1; j <= n; j++)
            scanf("%d",&c[i][j]);
        }
        for(i = 1; i <= n; i++)
        ta[i] = i;  // ta [i] = rand() % 101 也可以
        for(i = 1; i <= n; i++)
        {
            tc[i] = 0;
            tb[i] = 0;
            for(j = 1; j <= n; j++)
            {
                tc[i] += c[i][j] * ta[j];
                tb[i] += b[i][j] * ta[j];
            }
        }
        for(i = 1; i <= n; i++)
        {
            ta[i] = 0;
            for(j = 1; j <= n; j++)
            ta[i] += a[i][j] * tb[j];
        }
        //int flag = 1;
        for(i = 1; i <= n; i++)
        {
            if(tc[i] != ta[i])
            {
                //flag = 1;
                cout<<"NO\n";
                break;
            }
        }
        if(i == n + 1) cout<<"YES\n";
    }
    return 0;
}