一道小学题

题目链接：https://cn.vjudge.net/contest/218567#problem/C

自己的一直wa，就找了个代码，先写一下博客吧

#include <iostream>
#include <stdio.h>
using namespace std;
long long a[505][505];
long long sum1[505];
long long sum2[505];
int main()
{
    int n;
    cin>>n;
    if(n==1)
    {
        int t;
        cin>>t;
        printf("1\n");
        return 0;
    }
    long long flag1,flag2,flag,tep1,xx=0,gg=0;
    for(int i=1; i<=n; i++)
    {
        flag=0;
        for(int j=1; j<=n; j++)
        {
            scanf("%lld",&a[i][j]);
            if(a[i][j]==0)
            {
                flag=1;
                flag1=i;
                flag2=j;
            }
            sum1[i]+=a[i][j];//记录每一行数字的和
        }
        if(flag==0)
        {
            if(xx==0)
            {
                tep1=sum1[i];//tep1存放零所在行的数的和
                xx=1;
            }
            if(sum1[i]!=tep1)//如果没有零，用gg记录
                gg=1;
        }
    }
     printf("%d %d \n",flag1,flag2);
    if(gg==1)//没有零，输出-1
    {
        printf("-1\n");
        return 0;
    }
    long long tep2=0,yy=0;
    for(int j=1; j<=n; j++)
    {
        flag=0;
        for(int i=1; i<=n; i++)
        {
            if(a[i][j]==0)
            {
                flag=1;
            }
            sum2[j]+=a[i][j];
        }
        if(flag==0)
        {
            if(yy==0)
            {
                tep2=sum2[j];
                yy=1;
            }
            if(sum2[j]!=tep2)
                gg=1;
        }
    }
      printf("%d\n",tep2);
    if(gg==1)
    {
        printf("-1\n");
        return 0;
    }
    long long  ss=0;
    for(int j=1; j<=n; j++)
        ss+=a[flag1][j];
     printf("ss=%d\n",tep1);
    printf("%dgg\n",a[flag1][flag2]);
    a[flag1][flag2]=tep1-ss;//记录要输出的数
     printf("%d\n",a[flag1][flag2]);
    if(a[flag1][flag2]<=0)//所得数小于等于零不符合条件
    {
        printf("-1\n");
        return 0;
    }
    if(tep1==tep2)//行列相等
    {
        long long suma=0,sumb=0;
        for(int i=1; i<=n; i++)
            suma+=a[i][i];
        for(int i=1; i<=n; i++)
            sumb+=a[i][n-i+1];
printf("%lld %lld",suma,sumb);
        if(suma==sumb&&suma==tep1)
            printf("%lld\n",a[flag1][flag2]);
        else
            printf("-1\n");
    }
    else//行列不想等输出-1
        printf("-1\n");
    return 0;
}

自己的代码：

#include<math.h>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
long long map[505][505];
long long sum[260000];
long long sum1[50500];
int main()
{
    int n,k;
    long long s1,s2;
    scanf("%d",&n);
    if(n==1)
    {
        scanf("%lld",&k);
        printf("1\n");
    }
    else
    {

        memset(map,0ll,sizeof(map));
        memset(sum,0ll,sizeof(sum));
        memset(sum1,0ll,sizeof(sum1));
        s1=s2=k=0;
        int flag=0;
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                scanf("%lld",&map[i][j]);
                if(map[i][j]==0)
                {
                    flag=1;
                }
                sum[k]+=map[i][j];//记录每一行的和
                if(i==j)
                    s1+=map[i][j];//对角线的和
                if((i+j)==n-1)
                    s2+=map[i][j];
            }
            k++;
        }
        if(!flag)
        {
            printf("-1\n");
        }
        else
        {
            for(int j=0; j<n; j++)//记录每一列的和
            {
                for(int i=0; i<n; i++)
                {
                    sum[k]+=map[i][j];
                }
                k++;
            }
            sum[k++]=s1;
            sum[k++]=s2;
            sort(sum,sum+k);
            int l=0;
//            for(int i=0; i<k; i++)
//                printf("%lld ",sum[i]);
//            printf("\n");
            for(int i=0; i<k; i++)
            {
                for(int j=i; j<k; j++)
                {
                    if(sum[j]-sum[i]!=0)
                        sum1[l++]=sum[j]-sum[i];
                }
            }
            sort(sum1,sum1+l);
//        for(int i=0; i<l; i++)
//            printf("%lld ",sum1[i]);
//        printf("\n");
            if(sum[k-1]==sum[k-3]&&(sum[k-3]-sum[0])>0&&sum[0]==sum[1])
                printf("%lld\n",sum[k-1]-sum[0]);
            else
                printf("-1\n");
        }
    }
}

千辛万苦找出bug，大致就是，用数组存放每一行或者列的和，用两个数存放对角线和次对角线的和，并且记录零出现的位置，之后，找出输出的数，对行，列，对角线，次对角线比较，若相等输出，否则-1；

特殊数据：       

5            3        3                3
1 1 1 2 1    1 0 1    2016 2016 2016   1 2 2
1 1 1 1 1    2 1 2    2016 0    2016   2 2 1
1 1 0 1 1    2 1 2    2016 0    2016   0 1 2
1 1 1 1 1
1 1 1 1 1

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
long long a[505][550];
long long sum[550];
long long sum1[550];
long long tep,tep1,s,s1;
int x,y;
int xx,yy;
int main()
{
    int n;
    scanf("%d",&n);
    if(n==1)
    {
        long long k;
        scanf("%lld",&k);
        printf("1\n");
    }
    else
    {
        int flag=1;
        int flag1=1;
        s=s1=0;
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                scanf("%lld",&a[i][j]);
                if(a[i][j]==0)
                {
                    flag=0;
                    x=i,y=j;
                }
                sum[i]+=a[i][j];
                if(i==j)
                    s+=a[i][j];
                if((i+j)==n-1)
                    s1+=a[i][j];
            }
        }
        if(!flag)
        {
            tep=sum[x];
        }
        for(int j=0;j<n;j++)
        {
            for(int i=0;i<n;i++)
            {
                if(a[i][j]==0)
                {
                    flag1=0;
                    xx=i,yy=j;
                }
                sum1[j]+=a[i][j];
            }
        }
        tep1=sum1[yy];
        if(tep!=tep1)
        {
            printf("-1\n");
        }
        else
        {
            sort(sum,sum+n);
            sort(sum1,sum1+n);
//             printf("%lld %lld\n",tep,tep1);
            int f=1,ff=1;
            long long k=0,kk=0;
            k=sum[n-1]-tep;
            kk=sum1[n-1]-tep1;
            if(!f||!ff||k!=kk||kk<=0||k<=0||sum[n-1]!=sum[1]||sum1[n-1]!=sum1[1])
            {
                printf("-1\n");
            }
            else
            {
                if(x==y)
                    s+=k;
                if(x+y==n-1)
                    s1+=k;
                if(s1==s&&s==sum[n-1]&&s1==sum[n-1])
                    printf("%lld\n",k);
                else
                    printf("-1\n");
            }
        }
    }
}