Codeforces-808D Array Division (multiset 折半???)
Posted tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Codeforces-808D Array Division (multiset 折半???)相关的知识,希望对你有一定的参考价值。
题目链接:
http://codeforces.com/problemset/problem/808/D
题意:
给定一个数列,移动0或1个数字,使数列能从某个位置分开前后两半的和相等。
思路:
from: http://www.cnblogs.com/robin1998/p/6864278.html
我们可以假想有个隔板从左向右扫,每次两边的和的差值为x,若能找到某一边(看那边大)的一个数为x/2,则将它放到另一边就满足条件。扫一遍O(n),数据是无序的,如果暴力查找则总复杂度O(n^2)无法承受,我们可以维护两个平衡树做到O(nlogn),这里我们发现完全可以用STL的multiset代替。
菜鸡觉得是很奇妙的想法,这也是折半查找的思想把?
代码:
#include<bits/stdc++.h> using namespace std; typedef long long ll; #define MS(a) memset(a,0,sizeof(a)) #define MP make_pair #define PB push_back const int INF = 0x3f3f3f3f; const ll INFLL = 0x3f3f3f3f3f3f3f3fLL; inline ll read(){ ll x=0,f=1;char ch=getchar(); while(ch<\'0\'||ch>\'9\'){if(ch==\'-\')f=-1;ch=getchar();} while(ch>=\'0\'&&ch<=\'9\'){x=x*10+ch-\'0\';ch=getchar();} return x*f; } ////////////////////////////////////////////////////////////////////////// const int maxn = 1e5+10; int n,a[maxn]; multiset<ll> s1,s2; multiset<ll>::iterator it; int main(){ cin >> n; ll sum = 0; for(int i=1; i<=n; i++){ cin >> a[i]; sum += a[i]; s2.insert(a[i]); } if(sum%2) return 0*puts("NO"); ll sum1=0,sum2=sum; bool f = 0; for(int i=1; i<=n; i++){ if(sum1 == sum2) { f=1; break;} ll t = abs(sum1-sum2); if(sum1 > sum2){ if(t%2==0 && s1.count(t/2)){ f = 1; break; } }else{ if(t%2==0 && s2.count(t/2)){ f = 1; break; } } sum1 += a[i]; sum2 -= a[i]; s1.insert(a[i]); it = s2.find(a[i]); s2.erase(it); } if(f) puts("YES"); else puts("NO"); return 0; }
以上是关于Codeforces-808D Array Division (multiset 折半???)的主要内容,如果未能解决你的问题,请参考以下文章