CSP历年复赛题-P5016 [NOIP2018 普及组] 龙虎斗

原题链接：https://www.luogu.com.cn/problem/P5016

题意解读：计算在哪一号兵营增加s2名士兵，使得龙虎双方势力相差最小。

解题思路：

1、先根据初始情况，计算龙、虎双方各自的势力，包括p1号兵营增加s1个士兵

2、在枚举在1~n号兵营增加s2个士兵时，龙、虎双方的势力之差的最小值，如果是最小值则更新答案

3、输出答案，全程注意用long long

100分代码：

#include <bits/stdc++.h>
using namespace std;

long long n, m, p1, s1, s2;
long long c[100005];
long long l, r; //龙、虎两方的气势
long long minx = LLONG_MAX; //最小差值
int 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 < m; i++) l += (m - i) * c[i];
    for(int i = m + 1; i <= n; i++) r += (i - m) * c[i];
    //计算s2加到任意一队的气势差值最小时的队伍编号
    for(int i = 1; i <= n; i++)
    {
        long long nl = l, nr = r;
        if(i < m) nl += (m - i) * s2;
        if(i > m) nr += (i - m) * s2;
        if(abs(nl - nr) < minx)
        {
            minx = abs(nl - nr);
            ans = i;
        }
    }
    cout << ans;
    return 0;
}