Codeforces Round #521 (Div. 3) C - Good Array （前缀和）

题目链接：http://codeforces.com/contest/1077/problem/C

题意：一个长度为n的序列。删掉某个数字，使得序列最大值等于其他数字的总和。问有多少种情况。

题解：先解释一下样例。

5

2 5 1 2 2

如我们删掉某个2的话，就会有 5 = 2 + 1 + 2总有三种删法

题解：n是2e5，a【i】是1e6.暴力枚举没戏。

桶排记录每一位数字的位置。然后前缀和来表示除了 i 位置，其他数字总和sum。若想当前位置为最大，那么我们需要删掉       sum - a【i】的值，然后判断sum - a【i】在桶里面是否存在。存在就把这些位置加入结果当中。

一些细节需要注意。

1.所有值的和超int 所以 sum - a【i】的值也有可能爆int，但是我们每次最大才能删掉一个 1e6的数字，所以若sum-a【i】 大于1e6就直接GG。否则会re掉。

2.如 2 8 5 ，若想要5变成最大值，则需要删去一个5，但是序列只有一个5.就是它本身。没办法删去，所以需要判断要删掉的某个值的个数是否大于1.若大于1，就加入答案，等于1 就continue。

3，答案记得去重。

代码如下：

#include<bits/stdc++.h>
using namespace std;

const int maxn = 1e7 + 10;
int a[maxn];
vector<int> book[maxn]; 
set <int> ans;
long long pre[maxn];

int main(){
	int n ;
	scanf("%d",&n);
	ans.clear();
	for(int i = 1 ; i <= n ; i ++){
		scanf("%d",&a[i]);
		book[a[i]].push_back(i);
	}
	pre[1] = a[1];
	for(int i = 2; i <= n ; i ++){
		pre[i] = pre[i-1] + a[i];
	}
	int flag = 0;
	for(int i = 1 ; i <= n; i++){
		long long temp = pre[i - 1 ] ;
		temp += pre[n] - pre[i];
		
	//	cout << i << " " << temp << endl;
		if(temp < a[i]) continue;
		else if(temp == a[i]){
			flag = 1;
		}
		else {
			flag = 1;
			long long cha = temp - a[i];
			if(cha >= 2e6) continue;
			if(cha == a[i] && book[cha].size() == 1){
				continue;
			}
			if(book[cha].size() != 0 ){
				for(int j = 0 ; j < book[cha].size() ; j ++){
					ans.insert(book[cha][j]);
				}
			}
		}
	}	
	if(!flag){
			printf("0\n");
		}
	else{
			printf("%d\n",ans.size());
			set<int>::iterator it;
			for(it=ans.begin();it!=ans.end();it++) {
				printf("%d ",*it );
			}
			
			printf("\n");
		}
}

/*
5
2 5 1 2 2
6
1 2 3
4
8 3 5 2
5
2 1 2 4 3
*/