C++-POJ3213-PM3-[矩阵乘法]

已知矩阵乘法是n^3的，必然超时

故可以在需要验证的等式AB=C两边同时左乘D

一个1xN的任意的不含0矩阵

设E=DA，F=EB，G=DC，则此时只需验证F=G

当匹配到非法列J时，跳出n^2寻找行I即可

记录一下Ans，C[I,J]的正确值，然后就愉快地AC了

为了偷懒，我使用的矩阵是A~G连续编号的，所以可以放入数组，不喜勿喷！

#include <cstdio>
typedef long long ll;
const int MAXN=1005;
struct Matrix{ll a[MAXN][MAXN];}A[8];
void Mul(int a,int b,int c,int n,int p,int m){
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            for(int k=1;k<=p;k++)
                A[a].a[i][j]+=A[b].a[i][k]*A[c].a[k][j];
}
int main(){
    int N,P,M;
    while(scanf("%d%d%d",&N,&P,&M)!=EOF){
        for(int i=0;i<MAXN;i++)for(int j=0;j<MAXN;j++)for(int k=0;k<8;k++)A[k].a[i][j]=0;
        for(int i=1;i<=N;i++)for(int j=1;j<=P;j++)scanf("%lld",&A[1].a[i][j]);
        for(int i=1;i<=P;i++)for(int j=1;j<=M;j++)scanf("%lld",&A[2].a[i][j]);
        for(int i=1;i<=N;i++)for(int j=1;j<=M;j++)scanf("%lld",&A[3].a[i][j]);
        for(int i=1;i<=N;i++)A[4].a[1][i]=(i&1)?1:-1;//这里可以乱写，233 
        Mul(5,4,1,1,N,P),Mul(6,5,2,1,P,M),Mul(7,4,3,1,N,M);
        int flag=0,I,J;ll Ans;
        for(int i=1;i<=M;i++)if(A[6].a[1][i]!=A[7].a[1][i]){J=i;flag=1;break;}
        if(flag){
            for(int i=1;i<=N;i++){
                ll ans=0;
                for(int k=1;k<=P;k++)ans+=A[1].a[i][k]*A[2].a[k][J];
                if(ans!=A[3].a[i][J]){Ans=ans,I=i;break;}
            }
            printf("No
%d %d
%lld
",I,J,Ans);
        }
        else puts("Yes");
    }
    return 0;
}