C. Johnny and Another Rating Drop
Description
输入 n(2≤n≤2e5) 和长为 n 的数组 a(1≤a[i]≤1e6)。
对于数组 b，如果 b 中存在一个数 x，使得 x = b 中其余元素之和，则称 b 为「好数组」。
如果删除 a[i] 可以使剩余元素组成好数组，则称 i 为「好下标」。
输出 a 的好下标的个数，以及所有好下标（任意顺序）。注意下标从 1 开始。
Hints/Notes
Solution
Language: C++
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
| #include <iostream>
#include <utility>
#include <string>
#include <cstring>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <unordered_map>
#include <unordered_set>
#include <algorithm>
#include <numeric>
#include <fstream>
using namespace std;
istream &in = cin;
ostream &out = cout;
void solve(vector<int>& nums) {
long long max1 = 0, max2 = 0, maxIdx = 0, sum = 0;
for (int i = 0; i < nums.size(); i++) {
if (nums[i] > max1) {
max2 = max1;
max1 = nums[i];
maxIdx = i;
} else if (nums[i] > max2) {
max2 = nums[i];
}
sum += nums[i];
}
vector<int> res;
for (int i = 0; i < nums.size(); i++) {
int mx = (i == maxIdx) ? max2 : max1;
if (sum - nums[i] == 2 * mx) {
res.push_back(i + 1);
}
}
out << res.size() << endl;
for (auto v : res) {
out << v << " ";
}
}
int main() {
int n;
in >> n;
vector<int> nums;
for (int i = 0; i < n; i++) {
int num;
in >> num;
nums.push_back(num);
}
solve(nums);
return 0;
}
|