龙虎斗

题目描述

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

下面图1位n=6的实例：

图1.n=6的示例

轩轩在左侧，代表“龙”；凯凯在右侧，代表“虎”。他们以m号兵营作为分界，靠左的工兵属于龙势力，靠右的工兵属于虎势力，而第m号兵营中的工兵很纠结，他们不属于任何一方。

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

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

图2.n=6，m=4的示例

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

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

输入描述

输入文件名为fight.in。

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

输出描述
输出文件名为fight.out。

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

样例

输入

6
2 3 2 3 2 3
4 6 5 2
输出

2

提示

说明：见图2

双方以m=4号兵营分界，有s1=5位工兵突然出现在p1=6号兵营。

龙方的气势为：2×（4-1）+3×（4-2）+2×（4-3）=14

虎方的气势为：2×（5-4）+（3+5）×（6-4）=18

当你将手中的s2=2位工兵派往p2=2号兵营时，龙方的气势变为：14+2×（4-2）=18

此时双方气势相等。

1<m<n,1<=p1<=n。

对于20%的数据，n=3，m=2，ci=1，s1，s2<=100.

另有20%的数据，n<=10，p1=m,ci=1,s1,s2<100。

对于60%的数据，n<=100,ci=1,s1,s2<=100.

对于80%的数据，n<=100,ci,s1,s2=100

对于100%的数据，n<100000,ci,s1,s2<=1000000000。

思路：

（读完题后发现题目比较长，而且还没怎么看懂，但是若仔细看看，就不难发现，这道题就是一道模拟题，只要读懂了题目就可以做出来）

1）先把 s1 个兵添加上去 b[p1]+=s1；
2）然后计算龙虎双方的气势值；
3）枚举每个位置，尝试将 s2 个兵添加到这个位置，求出气势差值。
按每次求出的气势差值比较，得到最小的气势差值，以及气势差值最小时，将 s2 个
兵安排的位置。
注意：数据范围中，每个位置的工兵数最大为 10^9，共有 10^5 个位置，加起来的气势
和可以估算为 10^9* 10^5=10^15 超出了 int 型变量可以表达的范围，所以对某些变量
要用 long long 类型。
以样例为例加以说明：

下标            1          2          3          4          5          6

数组b          2          3          2          3          2          3

                                                       m

下标            1          2          3          4          5          6         p1=6

数组b          2          3          2          3          2          8

                                                                             s1=5

龙气势值：longg=2*(4-1)+3*(4-2)+2*(4-3)=14
虎气势值：hu=2*(5-4)+8*(6-4)=18
气势差值 minn=|14-18|=4
s2=2 应该放在什么位置？
如果将 s2 放置在 m 位置的左边营地某一个位置的情况

当 i=1 时，longg=long+s2*(4-1)=20
气势差值为：|20-18|=2<minn=4
气势差值 minn=2                                                                位置 p2=1

下标    1    2    3    4(m)    5    6
数组b    2    3    2    3    2    8

本次最后将 s2*(4-1)的值减去，保持 longg 还是未放置 s2 时的状态
当 i=2 时，longg=long+s2*(4-2)=18
气势差值为：|18-18|=0<minn=2
气势差值 minn=0                                                               位置 p2=2

下标    1    2    3    4(m)    5    6
数组b    2    3    2    3    2    8

本次最后将 s2*(4-2)的值减去，保持 longg 还是未放置 s2 时的状态
当 i=3 时，longg=long+s2*(4-3)=16
气势差值为：|16-18|=2>minn=0
不做任何操作，不改变 s2 放置的位置

下标    1    2    3    4(m)    5    6
数组b    2    3    2    3    2    8

本次最后将 s2*(4-3)的值减去，保持 longg 还是未放置 s2 时的状态
如果将 s2 放置在 m 位置的右边营地某一个位置的情况
当 i=5 时，hu=hu+s2*(5-4)=20
气势差值为：|20-16|=4>minn=0
不做任何操作，不改变 s2 放置的位置

下标    1    2    3    4(m)    5    6
数组b    2    3    2    3    2    8

本次最后将 s2*(5-4)的值减去，保持 longg 还是未放置 s2 时的状态
当 i=6 时，hu=hu+s2*(6-4)=22
气势差值为：|22-16|=6>minn=0
不做任何操作，不改变 s2 放置的位置

下标    1    2    3    4(m)    5    6
数组b    2    3    2    3    2    8
本次最后将 s2*(6-4)的值减去，保持 longg 还是未放置 s2 时的状态
最终输出 s2 放置的位置：p2=2

代码

#include<iostream>
#include<iomanip>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<stack>//用万能头文件也行
using namespace std;
int main(){
    long long n,m,p1,s1,s2;
    long long a[100005];
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    cin>>m>>p1>>s1>>s2;
    a[p1]+=s1;
    long long suml=0,sumh=0;
    for(int i=1;i<m;i++){
        suml+=a[i]*(m-i);
    }
    for(int i=m+1;i<=n;i++){
        sumh+=a[i]*(i-m);
    }
    long long minn=abs(suml-sumh);
    int pos=m;
    if(suml<sumh){
        for(int i=1;i<m;i++){
            if(abs(suml+s2*(m-i)-sumh)<minn){
                pos=i;
                minn=abs(suml+s2*(m-i)-sumh);
            }
        }
    }
    else if(sumh<suml){
        for(int i=m+1;i<=n;i++){
            if(abs(sumh+s2*(i-m)-suml)<minn){
                pos=i;
                minn=abs(sumh+s2*(i-m)-suml);
            }
        }
    }
    cout<<pos;
    return 0;
}