poj3213(特殊的矩阵乘法)

链接：点击打开链接

题意：给出N*P的矩阵a，P*M的矩阵b，N*M的矩阵c，c=a*b，但是c中可能有一个元素出现错误，找出错误的位置

代码：

#include <bitset>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int r1[1005],r2[1005];
int a[1005][1005],b[1005][1005],c[1005][1005];
int main(){                                     //对于矩阵乘法来说，复杂度大部分是N^3但是
    int i,j,k,n,m,p,id,tmp;                     //当一个矩阵只有一维时，则复杂度变为N^2，
    while(scanf("%d%d%d",&n,&p,&m)!=EOF){       //所以对于这道题，我们可以想到两边左乘一个
        for(i=1;i<=n;i++)                       //一维所有元素全是1的矩阵，再判断是否想相同
        for(j=1;j<=p;j++)
        scanf("%d",&a[i][j]);
        for(i=1;i<=p;i++)
        for(j=1;j<=m;j++)
        scanf("%d",&b[i][j]);
        for(i=1;i<=n;i++)
        for(j=1;j<=m;j++)
        scanf("%d",&c[i][j]);
        memset(r1,0,sizeof(r1));
        memset(r2,0,sizeof(r2));
        for(i=1;i<=p;i++){                      //就相当于第一个矩阵每列相加形成行矩阵后，再
            for(j=1;j<=n;j++)                   //乘第二个矩阵
            r1[i]+=a[j][i];
        }
        for(i=1;i<=m;i++){
            for(j=1;j<=p;j++)
            r2[i]+=r1[j]*b[j][i];
        }
        memset(r1,0,sizeof(r1));
        for(i=1;i<=m;i++){
            for(j=1;j<=n;j++)
            r1[i]+=c[j][i];
        }
        id=-1;
        for(j=1;j<=m;j++){                      //通过左右乘完后的一维矩阵判断哪列可能出现错误
            if(r1[j]!=r2[j]){
                id=j;
                break;
            }
        }
        if(id==-1)
        puts("Yes");
        else{
            for(i=1;i<=n;i++){                  //直接在列上寻找
                tmp=0;
                for(j=1;j<=p;j++)
                tmp+=a[i][j]*b[j][id];
                if(tmp!=c[i][id]){
                    puts("No");
                    printf("%d %d\n",i,id);
                    printf("%d\n",tmp);
                }
            }
        }
    }
    return 0;
}