711B - Chris and Magic Square 模拟

题目大意:在num[i][j]==0处填一个数使每行,每列,对角线的和相同,若果有多种答案输出一种。

题目思路:模拟

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<stdlib.h>
#include<queue>
#include<math.h>
#include<map>
#define INF 0xffffffff
#define MAX 505
#define Temp 1000000000
#define MOD 1000000007
#define LL long long
#define PI acos(-1) using namespace std; long long num[MAX][MAX]; long long change(int n,int x,int y)
{
long long k=,sum1=,sum2=,sum3=,sum4=,sum=;
int op1=,op2=;
for(int i=; i<=n; i++)
{
if(i!=x)
{
for(int j=; j<=n; j++)
k+=num[i][j];
break;
}
}
for(int i=; i<=n; i++)
{
sum=;
if(i==x)
{
for(int j=; j<=n; j++)
sum1+=num[i][j];
continue;
}
for(int j=; j<=n; j++)
{
sum+=num[i][j];
}
if(sum!=k)
return -;
} for(int i=; i<=n; i++)
{
sum=;
if(i==y)
{
for(int j=; j<=n; j++)
sum2+=num[j][i];
continue;
}
for(int j=; j<=n; j++)
{
sum+=num[j][i];
}
if(sum!=k)
return -;
}
if(x==y && x==(n+)/ && n%!=)
{
for(int i=; i<=n; i++)
{
sum3+=num[i][i];
sum4+=num[i][n-i+];
}
long long a=k-sum1;
long long b=k-sum2;
long long c=k-sum3;
long long d=k-sum4;
if(a!=b || a!=c || a!=d || b!=c || b!=d || c!=d)
return -;
else if(a<)
return -;
return a;
}
for(int i=; i<=n; i++)
{
if(num[i][i]==)
{
op1=;
}
sum3+=num[i][i];
if(num[i][n+-i]==)
{
op2=;
}
sum4+=num[i][n+-i];
}
if(op1== && sum3!=k)
return -;
if(op2== && sum4!=k)
return -;
if(op1)
{
long long c=k-sum3;
long long a=k-sum1;
long long b=k-sum2;
if(a!=b || a!=c || b!=c)
return -;
else if(a < )
return -;
return a;
}
if(op2)
{
long long c=k-sum4;
long long a=k-sum1;
long long b=k-sum2;
if(a!=b || a!=c || b!=c)
return -;
else if(a < )
return -;
return a;
}
long long a=k-sum1;
long long b=k-sum2;
if(a!=b || (a==b && a<))
return -;
return a;
} int main()
{
int n,x,y;
while(scanf("%d",&n)!=EOF)
{
memset(num,,sizeof(num));
for(int i=; i<=n; i++)
{
for(int j=; j<=n; j++)
{
scanf("%lld",&num[i][j]);
if(num[i][j]==)
{
x=i;
y=j;
}
}
}
if(n==)
{
printf("2\n");
continue;
}
long long ok=change(n,x,y);
printf("%lld\n",ok);
}
return ;
}
上一篇:网络流24题第一题(luogu2796飞行员配对方案)


下一篇:C语言库函数大全及应用实例八