Codeforces-808D Array Division multiset
Posted R o b i n
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Codeforces-808D Array Division multiset相关的知识,希望对你有一定的参考价值。
Description
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?
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.
Print YES if Vasya can divide the array after moving one element. Otherwise print NO.
3
1 3 2
YES
5
1 2 3 4 5
NO
5
2 2 3 4 5
YES
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.
题解:
题意是给定一个数列,移动0或1个数字,使数列能从某个位置分开前后两半的和相等。我们可以假象有个隔板从左向右扫,每次两边的和的差值为x,若能找到某一边(看那边大)的一个数为x/2,则将它放到另一边满足条件。扫一遍O(n),数据是无序的,如果暴力查找则总复杂度O(n^2)无法承受,我们可以维护两个平衡树做到O(nlogn),这里我们发现完全可以用STL的multiset代替。
代码:
1 #include<bits/stdc++.h> 2 using namespace std; 3 #define INF 0x3f3f3f3f 4 #define M(a, b) memset(a, b, sizeof(a)) 5 const int N = 100000 + 5; 6 long long a[N]; 7 multiset<long long> s1, s2; 8 multiset<long long>::iterator it; 9 10 int main() { 11 ios::sync_with_stdio(false); 12 int n; 13 while (cin >> n) { 14 s1.clear(); s2.clear(); 15 long long sum = 0; 16 for (int i = 1; i <= n; ++i) cin >> a[i], sum += a[i], s2.insert(a[i]); 17 if (sum & 1) {cout << "NO\n"; continue;} 18 long long sum1 = 0, sum2 = sum; 19 if (s2.count(sum/2)) {cout << "YES\n"; continue;} 20 bool flag = 0; 21 long long temp; 22 for (int i = 1; i <= n; ++i) { 23 if (sum1 == sum2) {flag = 1; break;} 24 if (sum1 < sum2) { 25 temp = sum2-sum1; 26 if (temp%2==0 && s2.count(temp/2)) {flag = 1; break;} 27 } 28 else { 29 temp = sum1-sum2; 30 if (temp%2==0 && s1.count(temp/2)) {flag = 1; break;} 31 } 32 sum1 += a[i]; s1.insert(a[i]); 33 sum2 -= a[i]; it = s2.find(a[i]); s2.erase(it); 34 } 35 if (flag) cout << "YES\n"; 36 else cout << "NO\n"; 37 38 } 39 40 return 0; 41 }
以上是关于Codeforces-808D Array Division multiset的主要内容,如果未能解决你的问题,请参考以下文章