关于八月POJ月赛的problem B(3381)的解题总结
一、 题目要求
给定三个矩阵:A、B和C,判断A*B是否等于C。
二、 分析
题目很直观,就是矩阵乘法,但是两个矩阵直接相乘需要O(n^3)的时间复杂度,假设A*B的结果存放在AB矩阵中,实例代码如下:
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
{
AB[i][j] = 0;
for(int k=0;k<n;k++)
AB[i][j] += A[i][k]*B[k][j];
}
然后在用O(n^2)的时间进行AB与C的比较,返回是否相等。比赛时很容易想到这个解法,但是超时是必然的,因为题目中矩阵的行列都不大于250,O(n^3)的时间算一个很大的矩阵相乘,要用超过2秒的时间。
怎么改进呢?
用压缩矩阵再比较的方法:
设一个行向量X,X是1*n的矩阵,若A*B等于C则必有X*A*B等于X*C,虽然多乘了一个向量,但是时间复杂度却降低到了O(n^2)了。
这样解的实质是把一个方阵压缩成了一个行向量,向量的每一个元素都是原矩阵该列的的和,也就是说用和来比较。这样大大节省了时间。但是带来一个问题:
如这两个矩阵:
x x x x x x x x x x x x x x x x x x x x
x x 1 x x x x 1 x x x x 0 x x x x 2 x x
x x x x x x x x x x x x x x x x x x x x
x x 1 x x x x 1 x x x x 2 x x x x 0 x x
…… ……
x x x x x x x x x x x x x x x x x x x x
用压缩再比较的方法不能得到正确结果。压缩再比较的关键在于怎么样在和中体现原来每个元素的个性。关键就是X行向量怎么设定。题目在比赛结束后提示是一个随机的向量X,就是把X的每个元素设为随机数。我觉得可以把X设为一个递增的向量:{1、2、…、n},这样更能体现每个元素的个性,而随机有可能在关键点上出现错误。
代码:
#include<iostream>
#include<string.h>
using namespace std;
__int64 a[510][510], b[510][510] , c[510][510];
__int64 temp1[510];
__int64 temp2[510];
__int64 temp3[510];
int n;
bool eql()
{
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
temp3[i]+=temp2[j]*b[j][i];
}
}
for(int i=0;i<n;i++)
if(temp1[i]!=temp3[i]) return false;
return true;
}
int main()
{
int t;
// scanf("%d",&t);
// while(t--)
while(scanf("%d",&n)!=EOF)
{
// scanf("%d",&n);
memset(temp1, 0, sizeof(temp1));
memset(temp2, 0, sizeof(temp2));
memset(temp3, 0, sizeof(temp3));
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++) scanf("%I64d",&a[i][j]);
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
scanf("%I64d",&b[i][j]);
temp2[i]+=(j+1)*a[j][i];
}
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++) scanf("%I64d",&c[i][j]);
}
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
temp1[i]+=(j+1)*c[j][i];
}
}
if(eql()) printf("YES\n");
else printf("NO\n");
}
return 0;
}