POJ 3318 两个大矩阵相乘是否与另一矩阵结果相等

关于八月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;
}