信息学奥赛一本通 1979：【18NOIP普及组】龙虎斗

题目

1979：【18NOIP普及组】龙虎斗

代码

注意long long求绝对值需要自己实现。

剩下不用多说了吧，大大的模拟。

#include<iostream>
#include<string>
#include<utility>
using namespace std;
typedef long long LL;
const int N = 1e5 + 14;
LL a[N];

// long long求绝对值
LL calc_abs(LL k) {
	return k < 0 ? -k : k;
}

// 求龙方和虎方的气势
pair<LL, LL> power(LL a[], LL n, LL m) {
	LL x = 0, y = 0; // x为龙方气势，y为虎方气势
	for (LL i = 1; i < m; i++) x += a[i] * (m - i);   // 龙方
	for (LL i = m + 1; i <= n; i++) y += a[i] * (i - m);  // 虎方
	return make_pair(x, y);
}

// 计算在第i兵营放s2位工兵后，双方气势差值
LL calc_delta(LL ql, LL qh, LL m, LL i, LL s2) {
	if (i < m) return calc_abs(ql + s2 * (m - i) - qh);
	return calc_abs(qh + s2 * (i - m) - ql);
}

int main() {
	LL n, m, p1, s1, s2;
	// 读入数据
	cin >> n;
	for (LL i = 1; i <= n; i++) cin >> a[i];
	cin >> m >> p1 >> s1 >> s2;
	// 天降神兵
	a[p1] += s1;

	// 计算气势
	LL ql, qh;
	pair<LL, LL> q = power(a, n, m);
	ql = q.first;
	qh = q.second;

	// delta存储当前气势差值，best存储最优气势差值，p2表示最优情况下放哪个兵营
	LL delta, best, p2;  // delta存储当前气势差值，best存储最小气势差值，p2表示最优情况下放哪个兵营

	// 差值最大就是当前差值（s2位工兵全派往m兵营）
	best = calc_abs(ql - qh);
	p2 = m;

	// 尝试将s2位工兵派往每个兵营
	for (LL i = 1; i <= n; i++) {
		delta = calc_delta(ql, qh, m, i, s2);  // 计算差值
		if (delta < best) {
			best = delta;
			p2 = i;
		}
	}
	cout << p2 << endl;
	return 0;
}

 评测结果