POJ3318 Matrix Multiplication

POJ3318 Matrix Multiplication

题目大意

给你三个$n\times n$的矩阵$A,B,C$，请你判断$A\times B$和$C$是否相等。

$1\le n\le 500$。

题解

随机生成一个向量$v$（也就是$1\times n$的矩阵），判断$v\times A\times B$与$v\times C$是否相等，如果相等，则$A\times B$有极大的概率等于$C$；若不相等，则$A\times B$一定不等于$C$。当然，如果不放心，可以多生成几个向量来提高正确性。

因为向量乘矩阵的时间复杂度为$O(n^2)$，所以不会TLE。

code

#include<iostream>
#include<cstdlib>
#include<time.h>
#include<cstdio>
using namespace std;
int n;
int v[505],t[505],w1[505],w2[505],a[505][505],b[505][505],c[505][505];
int main()
{
	srand(time(NULL));
	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]);
		}
	}
	for(int i=1;i<=n;i++){
		v[i]=rand()%100+1;
	}
	for(int j=1;j<=n;j++){
		for(int k=1;k<=n;k++){
			t[j]+=v[k]*a[k][j];
		}
	}
	for(int j=1;j<=n;j++){
		for(int k=1;k<=n;k++){
			w1[j]+=t[k]*b[k][j];
		}
	}
	for(int j=1;j<=n;j++){
		for(int k=1;k<=n;k++){
			w2[j]+=v[k]*c[k][j];
		}
	}
	for(int i=1;i<=n;i++){
		if(w1[i]!=w2[i]){
			printf("NO");
			return 0;
		}
	}
	printf("YES");
	return 0;
}

后记

在GDKOI的时候遇到了这道题，考后才发现是原题。但考场的数据比这个数据强，有模数，有多组数据，$\sum n\le 3000$。