题目链接: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
*/