题目链接：https://www.luogu.com.cn/problem/P5016
解题思路
首先要考虑 “某一刻天降神兵，共有 \(s_1\) 位工兵突然出现在了 \(p_1\) 号兵营” 这一干扰条件。
这 \(s_1\) 位工兵某一刻天降，和他们一开始就在兵营中是等价的！所以我直接把 \(s_1\) 加到 \(c_{p_1}\) 当中去即可。然后接下来我们就不用考虑 \(s_1\) 和 \(p_1\) 了。
龙队的气势和为
\[\sum_{i=1}^{m-1} c[i] \times (m - i)\]
虎队的气势和为
\[\sum_{i=m+1}^n c[i] \times (i-m)\]
两队气势和之差为（这里算的是虎队的气势和-龙队的气势和）
\[\sum_{i=1}^n c[i] \times (i-m)\]
而我如果在 \(p_2\) 位置放了 \(s_2\) 个工兵，则两队气势和之差会变成
\[s_2 \times (p_2 - m) + \sum_{i=1}^n c[i] \times (i-m)\]
所以我们只需要从 \(1\) 到 \(n\) 去枚举 \(p_2\)，然后找气势和之差的绝对值
\[| s_2 \times (p_2 - m) + \sum_{i=1}^n c[i] \times (i-m) |\]
最小的那个对应的就是 \(p_2\)。
实现代码如下：
#include <bits/stdc++.h>
using namespace std;
const int maxn = 100010;
int n, m, p1, p2;
long long c[maxn], s1, s2, sum, ans;
int main() {
cin >> n;
for (int i = 1; i <= n; i ++) cin >> c[i];
cin >> m >> p1 >> s1 >> s2;
c[p1] += s1;
for (int i = 1; i <= n; i ++) sum += c[i] * (i - m);
for (int i = 1; i <= n; i ++) {
long long tmp = abs(sum + s2 * (i - m));
if (!p2 || ans > tmp) {
p2 = i;
ans = tmp;
}
}
cout << p2 << endl;
return 0;
}