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