poj3213

    傳送門 http://poj.org/problem?id=3213

    題目大意：兩個矩陣(N<=1000)相乘，只有一個地方可能出錯，問是否有錯，並修改。

    矩陣乘法最低複雜度就是N^3的啊，要計算每一個的話，妥妥超時，但這題規定，只有可能一個地方出錯，可以通過求出前一個矩陣的列和，後一個矩陣的行和，進行簡化，十分巧妙。

      

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <iomanip>
#pragma comment(linker,"/STACK:102400000,102400000")

using namespace std;
#define   MAX           1005
#define   MAXN          100005
#define   maxnode       105
#define   sigma_size    2
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

const ull    prime = 999983;
const int    INF   = 0x3f3f3f3f;
//const LL     INFF  = 0x3f3f3f3f;
const double pi    = acos(-1.0);
const double inf   = 1e18;
const double eps   = 1e-9;
const LL     mod   = 1e9+7;
const ull    mxx   = 1333331;

/*****************************************************/
inline void RI(int &x) {
      char c;
      while((c=getchar())<'0' || c>'9');
      x=c-'0';
      while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
}
/*****************************************************/

int n,m,p,k;
int a[1010][1010],b[1010][1010],c[1010][1010];
LL suma[1010],sumb[1010],sumc1[1010],sumc2[1010];

int main(){
    while(cin>>n>>m>>k){
        mem(suma,0);
        mem(sumb,0);
        mem(sumc1,0);
        mem(sumc2,0);
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                scanf("%d",&a[i][j]);
                suma[j]+=a[i][j];
            }
        }
        for(int i=0;i<m;i++){
            for(int j=0;j<k;j++){
                scanf("%d",&b[i][j]);
                sumb[i]+=b[i][j];
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<k;j++){
                scanf("%d",&c[i][j]);
                sumc1[i]+=c[i][j];
                sumc2[j]+=c[i][j];
            }
        }
        bool bool1=true;
        int ans1=0,ans2=0;
        for(int i=0;i<n;i++){
            LL temp=0;
            for(int j=0;j<m;j++){
                temp+=((LL)a[i][j]*sumb[j]);
            }
            if(temp!=sumc1[i]){
                ans1=i;
                bool1=false;
                break;
            }
        }
        for(int i=0;i<k;i++){
            LL temp=0;
            for(int j=0;j<m;j++){
                temp+=((LL)b[j][i]*suma[j]);
            }
            if(temp!=sumc2[i]){
                ans2=i;
                break;
            }
        }
        if(bool1){
            cout<<"Yes"<<endl;
        }
        else {
            cout<<"No"<<endl;
            cout<<ans1+1<<' '<<ans2+1<<endl;
            LL ans=0;
            for(int i=0;i<m;i++){
                ans+=(LL)a[ans1][i]*b[i][ans2];
            }
            cout<<ans<<endl;
        }
    }
    return 0;
}