2450553_ac_405ms_3028k.cpp

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

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

int x[501];
int a[501][501], b[501][501], c[501][501];
int bx[501], ab[501], cx[501];

int n;

void init()
{
	int i;

	for(i = 0; i < n; i++)
		x[i] = (rand()%101)*(rand()%2==0?1:-1);
}

int main()
{
	int no = 1;
	int i, j;

	scanf("%d",&n);
	for(i = 0; i < n; i++)
		for(j = 0; j < n; j++)
			scanf("%d",&a[i][j]);
	for(i = 0; i < n; i++)
		for(j = 0; j < n; j++)
			scanf("%d",&b[i][j]);
	for(i = 0; i < n; i++)
		for(j = 0; j < n; j++)
			scanf("%d",&c[i][j]);
	while(no--)
	{
		init();
		memset(ab,0,sizeof(ab));
		memset(bx,0,sizeof(bx));
		memset(cx,0,sizeof(cx));
		for(i = 0; i < n; i++)
			for(j = 0; j < n; j++)
				bx[i] += b[i][j] * x[j];
		for(i = 0; i < n; i++)
			for(j = 0; j < n; j++)
				ab[i] += a[i][j] * bx[j];
		for(i = 0; i < n; i++)
		{
			for(j = 0; j < n; j++)
				cx[i] += c[i][j] * x[j];
			if(cx[i]!=ab[i])
			{
				printf("NO\n");
				return 1;
			}
		}
	}
	printf("YES\n");
	return 1;
}