AcWing 474. 龙虎斗

轩轩和凯凯正在玩一款叫《龙虎斗》的游戏，游戏的棋盘是一条线段，线段上有 𝑛 个兵营（自左至右编号 1∼𝑛），相邻编号的兵营之间相隔 1 厘米，即棋盘为长度为 𝑛−1 厘米的线段，𝑖 号兵营里有 c𝑖 位工兵。

下面图为 𝑛=6 的示例：

轩轩在左侧，代表“龙”；凯凯在右侧，代表“虎”。 

他们以 m 号兵营作为分界，靠左的工兵属于龙势力，靠右的工兵属于虎势力，而第 m 号兵营中的工兵很纠结，他们不属于任何一方。  

一个兵营的气势为：该兵营中的工兵数 × 该兵营到 m 号兵营的距离；参与游戏一方的势力定义为：属于这一方所有兵营的气势之和。

下面图为 n=6,𝑚=4 的示例，其中红色为龙方，黄色为虎方：

游戏过程中，某一刻天降神兵，共有 𝑠1 位工兵突然出现在了 𝑝1 号兵营。

作为轩轩和凯凯的朋友，你知道如果龙虎双方气势差距太悬殊，轩轩和凯凯就不愿意继续玩下去了。

为了让游戏继续，你需要选择一个兵营 𝑝2，并将你手里的 𝑠2 位工兵全部派往兵营 𝑝2，使得双方气势差距尽可能小。  

注意：你手中的工兵落在哪个兵营，就和该兵营中其他工兵有相同的势力归属（如果落在 m 号兵营，则不属于任何势力）。

输入格式

输入文件的第一行包含一个正整数 𝑛，代表兵营的数量。  

接下来的一行包含 𝑛 个正整数，相邻两数之间以一个空格分隔，第 𝑖 个正整数代表编号为 𝑖 的兵营中起始时的工兵数量 𝑐𝑖。  

接下来的一行包含四个正整数，相邻两数间以一个空格分隔，分别代表 𝑚,𝑝1,𝑠1,𝑠2。

输出格式

输出文件有一行，包含一个正整数，即 𝑝2，表示你选择的兵营编号。

如果存在多个编号同时满足最优，取最小的编号。

数据范围

1<m<n 
1≤p1≤n 
n≤105,ci,s1,s2≤109 

输入样例：

6 
2 3 2 3 2 3
4 6 5 2

输出样例：

2

#include<bits/stdc++.h>
using namespace std;
long long m,p1,s1,s2,fi[1000005];
int n;  
int main()
{
    long long MIN=1e19,w;
    long long su1=0,su2=0;
    long long k1,k2;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%lld",&fi[i]);
    scanf("%lld %lld %lld %lld",&m,&p1,&s1,&s2);
    fi[p1]+=s1;
    for(int i=1;i<=n;i++){
        if(i<m)
            su1+=(m-i)*fi[i];
        else if(i>m)
            su2+=(i-m)*fi[i];
    }       
    for(int i=1;i<=n;i++){
        k1=su1;
        k2=su2;
        if(i<m)
            k1+=(m-i)*s2;
        else if(i>m)
            k2+=(i-m)*s2;
        long long tep=abs(k1-k2);
        if(tep<MIN){
            MIN=tep;
            w=i;
        }
    }
    printf("%lld",w);
    return 0;
}