POJ 3318 Matrix Multiplication（随机算法）

Description
给出三个方阵A,B,C，判断A*B是否等于C
Input
第一行为一整数n表示方阵大小，之后为三个n*n矩阵分别表示A,B和C(n ≤ 500)
Output
如果A*B=C则输出YES，否则输出NO
Sample Input
2
1 0
2 3
5 1
0 8
5 1
10 26
Sample Output
YES
Solution
直接乘显然会爆，所以要用到随机化算法，这道题有两种做法
第一种是随机两个小于等于n的正整数x和y，用A的第x行乘B的第y列判断与C[x][y]是否相同，如此随机个几万次如果没有冲突就输出YES，否则输出NO
第二种是随机n个数作为一个列向量左乘A之后再左乘B，然后将得到的列向量与这个向量左乘C所得到的列向量相比较，相同则输出YES，否则输出NO
这里采用的是第一种方法
Code

#include<stdio.h>
#include<algorithm>
#include<time.h>
using namespace std;
#define maxn 501
int n,a[maxn][maxn],b[maxn][maxn],c[maxn][maxn];
int main()
{
    srand((unsigned)time(0));
    while(~scanf("%d",&n))
    {
        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]);
        int flag=1;
        for(int i=1;i<=40001&&flag;i++)
        {
            int x,y;
            int temp=0;
            x=rand()%n+1;
            y=rand()%n+1;
            for(int k=1;k<=n;k++)
                temp+=a[x][k]*b[k][y];
            if(temp!=c[x][y])flag=0;
        }   
        if(flag)printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}