c++龙虎斗

题目描述

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

下面图1为 n=6的示例∶

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

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

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

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

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

【数据规模与约定】

1<m<n,1≤P1≤n。

对于20%的数据，n=3，m=2，ci=1，si，S2≤100。

另有20%的数据，n≤10，p1=m，cl=1，sy，S2≤100。

对于60%的数据，n≤100，ct=1，Si，S2≤100。

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

对于100%的数据，n≤105，ci，S1S2≤109。

输入格式

输入文件名：fight.in

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

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

接下来的一行包含四个正整数，相邻两数间以一个空格分隔，分别代表m，Pl，Sy，S20

输出格式

输出文件名：fight.out

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

输入/输出例子1

输入：

6

2 3 2 3 2 3

4 6 5 2

输出：

2

输入/输出例子2

输入：

6

1 1 1 1 1 16

5 4 1 1

输出：

1

样例解释

【输入输出样例1说明】

见问题描述中的图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

当你将手中的 sz=2 位工兵派往 p2 =2 号兵营时，龙方的气势变为∶

14+2×(4-2)=18 此时双方气势相等。

【输入输出样例2说明】

双方以m=5号兵营分界，有s1=1位工兵突然出现在p1=4号兵营。龙方的气势为

1×(5-1)＋1×(5-2)＋1×(5-3)＋(1＋1)×(5-4)=11 虎方的气势为∶

16×(6-5)=16

当你将手中的 sz=1 位工兵派往 p2=1 号兵营时，龙方的气势变为∶

11＋1×(5-1)=15

此时可以使双方气势的差距最小。

#include<bits/stdc++.h>
using namespace std;  
long long n,a[1000000],m,p,s,k,sum,ans;
int main(){
    freopen("fight.in","r",stdin);
    freopen("fight.out","w",stdout);
    scanf("%lld",&n);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
    scanf("%lld%lld%lld%lld",&m,&p,&s,&k);
    for(int i=1;i<=n;i++) sum+=a[i]*(m-i);
    sum+=s*(m-p);
    ans=m+int(sum*1.0/k+0.5*(sum>0?1:-1)); 
    if(ans>n) ans=n;
    if(ans<1) ans=1;
    printf("%lld",ans);
    return 0;
}