题目信息

题目传送门

解题思路

大家看完题肯定能想到，我们其实把兵营的位置枚举一遍依次判断就可以得到答案了。但其实这是可以直接算出来哒~
两边兵营的气势分别为sum1，sum2，且保证sum1 < sum2（否则调换）,要将工兵放在sum1一边，那么我们要保证：
abs(sum1 + 工兵的数量 * L_{工兵，m号}, sum2)最小，不妨设左右式相等（①式）。
简化①式可得：
L_{工兵，m号} = （sum2 - sum1）/ 工兵数量。
我们只需借助此式算出答案并四舍五入即可。

代码实现

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 5;
typedef long long ll;
ll a[N];
int main() {
	int n;
    cin >> n;
    for (int i = 1; i <= n; ++i) {
		cin >> a[i];
	}
	ll m, s, p, k;
    cin >> m >> p >> s >> k;
	ll sum = 0, res = 0;
	for (int i = 1; i <= n; ++i) {
		sum += a[i] * (m - i);
	}
	sum += s * (m - p);
    res = m + (int) (sum * 1.0 / k + 0.5 * (sum > 0 ? 1 : -1)); // 注意特殊处理负数
    // 注意边界，答案必须在1~n范围内
	if (res > n) {
        res = n;
    }
	if (res < 1) {
        res = 1;
    }
	cout << res << '\n';
	return 0;
}