CodeForces - 711B Chris and Magic Square (暴力/模拟)

探讨了在给定n*n矩阵中找到合适的数字填充空白,使矩阵的行、列和两条对角线的和相等的问题。通过优化算法避免了暴力枚举,实现了高效的解决方案。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

       题目大意:输入 n ,输入 n*n 的矩阵,有一个占位 0 , 求得将 0 位置换成其他的整数 使得矩阵 行列斜 和全部相等。

       题解:一开始居然无脑地想用最暴力的办法,枚举空位可以填的数字,然后每次判断填上这个数字可不可行

这样的话时间复杂度是n^{2}*2n^{2},接近1e10了,肯定会T

我们可以先统计出空位在的那一行的和_sum,再随便找一行求和sum1,这样sum1-_sum就应该是我们要填入的数字,然后判断填入这个数字可不可以,可以就输出,不可以就是-1

但这个题有些坑,其实说是坑,实际是自己思维不够周全

①n=1的情况

那个这一个数肯定是0,那么我们最小就是填1

②要填的数字x\leq 0的情况

是不行的,要判断一下

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll a[600][600];
int n;
bool pd()
{
    ll value=-1;
    for(int i=1; i<=n; ++i)
    {
        ll he=0;
        for(int j=1;j<=n; ++j)
            he+=a[i][j];
        if(value==-1)
            value=he;
        else 
        if(value!=he)
            return 0;
    }
    for(int i=1; i<=n; ++i)
    {
        ll he=0;
        for(int j=1; j<=n; ++j)
            he+=a[j][i];
        if(value!=he)
            return 0;
    }
    ll he=0;
    for(int i=1; i<=n; ++i)
    {
        he+=a[i][i];
    }
    if(value!=he)
        return 0;
    he=0;
    for(int i=1; i<=n; ++i)
        he+=a[i][n-i+1];
    if(value!=he)
        return 0;

    return 1;
}
int main()
{
    //freopen("input.txt","r",stdin);
    cin>>n;
    int dx=-1,dy=-1;
    for(int i=1; i<=n; ++i)
        for(int j=1; j<=n; ++j)
        {
            cin>>a[i][j];
            if(a[i][j]==0)
                dx=i,dy=j;
        }
    if(n==1)
    {
        return cout<<1<<endl,0;
    }
    ll _sum=0,sum1=0;
    for(int i=1; i<=n; ++i)
        _sum+=a[dx][i];
    if(dx==1)
        for(int i=1; i<=n; ++i)
            sum1+=a[2][i];
    else for(int i=1; i<=n; ++i)
            sum1+=a[1][i];
    ll x=sum1-_sum;
    a[dx][dy]=x;
    if(pd() && x>0)
        cout<<x<<endl;
    else
        puts("-1");
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值