龙虎斗C语言

题目描述

轩轩和凯凯正在玩一款叫《龙虎斗》的游戏，游戏的棋盘是一条线段，线段上有 nn 个兵营（自左至右编号 1 \sim n1∼n），相邻编号的兵营之间相隔 11 厘米，即棋盘为长度为 n-1n−1 厘米的线段。ii 号兵营里有 c_ici
​位工兵。 下面图 1 为 n=6n=6 的示例：

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

一个兵营的气势为：该兵营中的工兵数\times× 该兵营到 mm 号兵营的距离；参与游戏 一方的势力定义为：属于这一方所有兵营的气势之和。
下面图 2 为 n = 6,m = 4n=6,m=4 的示例，其中红色为龙方，黄色为虎方：

s_1s1
​ 位工兵突然出现在了 p_1p1
​ 号兵营。作为轩轩和凯凯的朋友，你知道如果龙虎双方气势差距太悬殊，轩轩和凯凯就不愿意继续玩下去了。为了让游戏继续，你需要选择一个兵营 p_2p2
​，并将你手里的 s_2s2
​ 位工兵全部派往 兵营 p_2p2
​，使得双方气势差距尽可能小。

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

m

输入

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

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

接下来的一行包含四个正整数，相邻两数间以一个空格分隔，分别代表 m,p_1,s_1,s_2m,p1
​,s
​1,s2
​。

输出

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

样例输入

6 
2 3 2 3 2 3 
4 6 5 2 

样例输出

2

提示

 【数据规模与约定】

1 <m <n, 1 ≤ p1≤n。

对于20%的数据，n=3,m=2, ci= 1,s1, s2≤100。

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

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

对于80%的数据，n≤100, ci,s1,s2≤100。

对于100%的数据，n≤105,ci, s1, s2≤109。

#include<stdio.h>
long long i,j,k,n,m,p1,s1,s2;
long long a[100010],x,y;//注意数据提示的时候说了有百分之20会很大，所以用long long
long long abs(long long x1, long long y1) {
    if(x1 >= y1) return x1 - y1;
    return y1 - x1;
}
int main(){
    scanf("%lld",&n);
    for(i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    scanf("%lld%lld%lld%lld",&m,&p1,&s1,&s2);//以上为输入
    a[p1]+=s1;//在p1神兵天降s1人
    for(i=1;i<=n;i++){
        if(i<m)x+=(m-i)*a[i];
        if(i>m)y+=(i-m)*a[i];
    }//计算轩轩、凯凯方士气
    long long ans=1e22,num=1e22;//先最大化
    for(i=1;i<=n;i++){//枚举派兵位置
        if(i<m)x+=(m-i)*s2;//派给轩轩
        if(i>m)y+=(i-m)*s2;//派给凯凯
        long long u=abs(x,y);//计算轩轩、凯凯之差
        if(u<ans){
            ans=u;
            num=i;
        }//若此时枚举的士气差小于最小值，更新答案
        if(i<m)x-=(m-i)*s2;//复原
        if(i>m)y-=(i-m)*s2;//复原
    }
    printf("%d",num);//输出
    return 0;
}