Codeforces-808D Array Division (multiset 折半???)
生活随笔
收集整理的這篇文章主要介紹了
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; }?
轉載于:https://www.cnblogs.com/yxg123123/p/7226642.html
總結
以上是生活随笔為你收集整理的Codeforces-808D Array Division (multiset 折半???)的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: [Design-Pattern]工厂模式
- 下一篇: 【bzoj1705】[Usaco2007