Codeforces Round #521 (Div. 3) C. Good Array 思维

题解

给你一个n数字的序列 从序列中删除某些元素让剩下的序列当中有一个数等于其它数字之和

记录每个数字出现的次数 枚举每一个数字 尝试减去 如果原数列出现过减去之后和的一半则满足条件(需要排除自身a[i] == s2 / 2)

AC代码

#include <stdio.h>
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

const int INF = 0x3f3f3f3f;
const int MAXN = 1e6 + 10;
int a[MAXN], b[MAXN];

int main()
{
#ifdef LOCAL
	freopen("C:/input.txt", "r", stdin);
#endif
	int N;
	cin >> N;
	ll s = 0;
	for (int i = 1; i <= N; i++)
	{
		scanf("%I64d", &a[i]);
		b[a[i]]++;
		s += a[i];
	}
	vector<int> ans;
	for (int i = 1; i <= N; i++)
	{
		ll s2 = s - a[i]; //减去当前数字剩下的和
		if (s2 > 2e6 || s2 % 2) //和大于2e6 奇数则不可能分两份
			continue;
		if (b[s2 / 2] - (a[i] == s2 / 2) > 0) //否则判断是否有一个数字等于一半
			ans.push_back(i);
	}
	cout << ans.size() << endl;
	for (int i = 0; i < ans.size(); i++)
	{
		if (i)
			putchar(' ');
		printf("%d", ans[i]);
	}
	cout << endl;

	return 0;
}