J. n阶幻方矩阵验证

142 阅读1分钟
J. n阶幻方矩阵验证(数组)
题目描述

n阶幻方矩阵是指将123,...,n²数字填入n*n矩阵中,每行和,每列和,两条对角线的和均相等。给定n阶矩阵,验证是否n阶幻方矩阵。


输入

测试次数t

每组测试数据为:矩阵大小n,后跟n*n的矩阵数据。


输出

对每组测试数据,输出判定结果,YESNO


样例查看模式 
正常显示
查看格式
输入样例1 
3
3
8 1 6 
3 5 7
4 9 2
4
1   2 15 16
12 14  3  5
13  7 10  4
8  11  6  9
3
1 2 3
4 5 6
7 8 9

输出样例1
YES
YES
NO
#include<stdio.h>
int main()
{
	int t,n,i,j,vic=0,sum=0,w=0,h=0,up=0,down=0,num[50][50];
	scanf("%d",&t);
	while(t--)
	{
		vic=0;sum=0;w=0;h=0;up=0;down=0;
		scanf("%d",&n);
		for(i=0;i<n;i++)
		{
			for(j=0;j<n;j++)
			{
				scanf("%d",&num[i][j]);
				sum+=num[i][j];
			}
		}
		for(i=0;i<n;i++)
		{
			h+=num[i][0];
		}
		if((sum/n)==h)
		vic++;
		for(j=0;j<n;j++)
		{
			w+=num[0][j];
		}
		if((sum/n)==w)
		vic++;
		for(i=0;i<n;i++)
		{
			for(j=0;j<n;j++)
			{
				if(i==j)
				{
					down+=num[i][j];
				}
			}
		}
		if((sum/n)==down)
		vic++;
		for(i=0;i<n;i++)
		{
			for(j=0;j<n;j++)
			{
				if((i+j)==(n-1))
				{
					up+=num[i][j];
				}
			}
		}
		if((sum/n)==up)
		vic++;
		if(vic==4)
		{
			printf("YES\n");
		}
		else
		{
			printf("NO\n");
		}
	}
	return 0;
 } 

这里在每一行、每一列的和是否相等时我们使用了一个小技巧,就是如果你每一行的和、每一列的和都相等的话,那么你每一行的和、每一列的和都等于整个矩阵的和除以行数或者列数~~~~~~~~

判断对角线的时候也是如此~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~