洛谷：P5016 龙虎斗 ← NOIP2018 普及组T2

【题目来源】
https://www.luogu.com.cn/problem/P5016

【题目描述】
轩轩和凯凯正在玩一款叫《龙虎斗》的游戏，游戏的棋盘是一条线段，线段上有 n 个兵营（自左至右编号 1∼n），相邻编号的兵营之间相隔 1 厘米，即棋盘为长度为 n−1 厘米的线段。i 号兵营里有 ci 位工兵。 下面图 1 为 n=6 的示例：

轩轩在左侧，代表“龙”；凯凯在右侧，代表“虎”。他们以 m 号兵营作为分界，靠左的工兵属于龙势力，靠右的工兵属于虎势力，而第 m 号兵营中的工兵很纠结，他们不属于任何一方。
一个兵营的气势为：该兵营中的工兵数 × 该兵营到 m 号兵营的距离；参与游戏 一方的势力定义为：属于这一方所有兵营的气势之和。
下面图 2 为 n=6,m=4 的示例，其中红色为龙方，黄色为虎方：

游戏过程中，某一刻天降神兵，共有 s1 位工兵突然出现在了 p1 号兵营。作为轩轩和凯凯的朋友，你知道如果龙虎双方气势差距太悬殊，轩轩和凯凯就不愿意继续玩下去了。为了让游戏继续，你需要选择一个兵营 p2，并将你手里的 s2 位工兵全部派往兵营 p2，使得双方气势差距尽可能小。
注意：你手中的工兵落在哪个兵营，就和该兵营中其他工兵有相同的势力归属（如果落在 m 号兵营，则不属于任何势力）。

【输入格式】
输入文件的第一行包含一个正整数n，代表兵营的数量。
接下来的一行包含 n 个正整数，相邻两数之间以一个空格分隔，第 i 个正整数代表编号为 i 的兵营中起始时的工兵数量 ci。
接下来的一行包含四个正整数，相邻两数间以一个空格分隔，分别代表 m,p1,s1,s2。

【输出格式】
输出文件有一行，包含一个正整数，即 p2，表示你选择的兵营编号。如果存在多个编号同时满足最优，取最小的编号。

【算法代码】

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

const int maxn=1e5+5;
typedef long long ll;
ll n,m,p1,p2,s1,s2,sum,a[maxn];

int main() {
    cin>>n;
    for(int i=1; i<=n; i++) cin>>a[i];
    cin>>m>>p1>>s1>>s2;

    a[p1]+=s1;
    p2=1;
    for(int i=1; i<=n; i++) {
        sum+=(i-m)*a[i];
    }

    for(int i=1; i<=n; i++) {
        if(abs(sum+(i-m)*s2)<abs(sum+(p2-m)*s2))
            p2=i;
    }

    cout<<p2<<endl;

    return 0;
}

/*
input:
6
2 3 2 3 2 3
4 6 5 2

output:
2
-----------------
input:
6
1 1 1 1 1 16
5 4 1 1

output:
1
*/