CF808A前后缀+二分

题目

题目链接

给一个序列,问你能否移动一个元素的位置,使得存在某个位置,将序列分成左右两部分,他们的和相同。

题解

处理一下前缀和后缀,对于一个元素 a i a_i ai,如果在区间 [ i + 1 , n ] [i+1,n] [i+1,n]以内,存在 s u f f i x [ j ] + a i = = s u m / 2 suffix[j]+a_i==sum/2 suffix[j]+ai==sum/2,那么只需将 a i a_i ai移动到后面即可。前缀同理。由于前后缀具有单调性,可以二分实现。

tips:lower_bound在降序数组中,第四个参数填上greater<int>()即可。

代码
#include <bits/stdc++.h>
using namespace std;

long long n;
long long a[100010];
long long prefix[100010], suffix[100010];
int main()
{
    cin >> n;
    long long sum = 0;
    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        sum += a[i];
        prefix[i] = prefix[i - 1] + a[i];
    }
    if (sum & 1)
        return cout << "NO" << endl, 0;
    sum /= 2;
    bool flag = 0;
    for (int i = n; !flag && i >= 1; i--)
    {
        int pos = lower_bound(prefix, prefix + i, sum - a[i]) - prefix;
        if (pos >= 0 && pos < i && prefix[pos] + a[i] == sum)
            flag = 1;
        suffix[i] = suffix[i + 1] + a[i];
    }
    for (int i = 1; !flag && i <= n; i++)
    {
        int pos = lower_bound(suffix + 1, suffix + n + 2, sum - a[i], greater<long long>()) - suffix;
        if (pos > i && pos <= n + 1 && suffix[pos] + a[i] == sum)
            flag = 1;
    }
    if (flag)
        puts("YES");
    else
        puts("NO");
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

hesorchen

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值