D - Array Division

Vasya has an array a consisting of positive integer numbers. Vasya wants to divide this array into two non-empty consecutive parts (the prefix and the suffix) so that the sum of all elements in the first part equals to the sum of elements in the second part. It is not always possible, so Vasya will move some element before dividing the array (Vasya will erase some element and insert it into an arbitrary position).

Inserting an element in the same position he was erased from is also considered moving.

Can Vasya divide the array after choosing the right element to move and its new position?

Input

The first line contains single integer n (1 ≤ n ≤ 100000) — the size of the array.

The second line contains n integers a1, a2... an (1 ≤ ai ≤ 109) — the elements of the array.

Output

Print YES if Vasya can divide the array after moving one element. Otherwise print NO.

Example
Input
3
1 3 2
Output
YES
Input
5
1 2 3 4 5
Output
NO
Input
5
2 2 3 4 5
Output
YES
Note

In the first example Vasya can move the second element to the end of the array.

In the second example no move can make the division possible.

In the third example Vasya can move the fourth element by one position to the left.




这个题想要解决不难,双重循环暴力搜索可以出结果,但是会超时,因此我们需要优化,可以将sum[x]

的搜索化为二分,这样可以减少时间复杂度,因为sum是递增的嘛

#include<iostream>
using namespace std;

typedef long long ll;

const int maxn=100005;

int main()
{
    int n;
    ll a[maxn],sum[maxn];
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
        if(i==0)sum[i]=a[i];
        else sum[i]=sum[i-1]+a[i];
    }
    if(sum[n-1]%2==1)
    {
        cout<<"NO"<<endl;
        return 0;
    }
    ll d=sum[n-1]/2;
    for(int i=0;i<n;i++)
    {
        int l=0,r=n-1,mid;
        while(l<=r)
        {
            mid=(l+r)/2;
            if(sum[mid]-a[i]==d)
            {
                if(mid>i){cout<<"YES"<<endl;return 0;}
                else break;
            }
            if(sum[mid]-a[i]>d)r=mid-1;
            if(sum[mid]-a[i]<d)l=mid+1;

        }

        l=0;r=n-1;
        while(l<=r)
        {
            mid=(l+r)/2;
            if(sum[mid]+a[i]==d)
            {
                if(mid<i){cout<<"YES"<<endl;return 0;}
                else break;
            }
            if(sum[mid]+a[i]>d)r=mid-1;
            if(sum[mid]+a[i]<d)l=mid+1;
        }

    }
    cout<<"NO"<<endl;
    return 0;
}


  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

水能zai舟

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

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

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

打赏作者

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

抵扣说明:

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

余额充值