2444239_wa.cpp

北大大牛代码 1240道题的原代码 超级权威

#include <iostream>
#define MAX 501
using namespace std;
int N, P, M, is;
int i, j;
int A[MAX][MAX], B[MAX][MAX], x = 0, y = 0;
int C, Bp[MAX], Ap[MAX];
int temp, temp64, templ[MAX];


int main()
{
    
    scanf("%d", &N);
	P = N;M = N;
    is = 1;
    for (i = 0; i < P; i++)
    {
        Ap[i] = 0;
        templ[i] = 0;
    }
    for (i = 0; i < N; i++)
    {
        for (j = 0; j < P; j++)
        {
            scanf("%d", &A[i][j]);
            Ap[j] += A[i][j];
        }
    }
    for (i = 0; i < P; i++)
    {
        Bp[i] = 0;
        for (j = 0; j < M; j++)
        {
            scanf("%d", &B[i][j]);
            Bp[i] += B[i][j];
        }
    }
    for (i = 0; i < N; i++)
    {
        temp = 0;
        temp64 = 0;
        for (j = 0; j < P; j++)
        {
            temp += A[i][j] * Bp[j];
            if (j < M)
            {
                scanf("%ld", &C);
                temp64 += C;
                templ[j] += C;
            }
        }
        for (j = P; j < M; j++)
        {
            scanf("%ld", &C);
            temp64 += C;
            templ[j] += C;
        }
        if (temp64 != temp)
        {
            x = i;
            is = 0;
        }
    }
    for (i = 0; i < M; i++)
    {
        temp = 0;
        for (j = 0; j < P; j++)
        {
            temp += B[j][i] * Ap[j];
        }
        if (templ[i] != temp)
        {
            y = i;
            is = 0;
        }
    }
    if (is == 1)
        cout << "YES" << endl;
    else
		cout << "NO" <<endl;
	return 0;
}