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 ;
}