龙虎斗(打卡)

题目描述

轩轩和凯凯正在玩一款叫《龙虎斗》的游戏，游戏棋盘时一条线段，线段上有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。

#include<bits/stdc++.h>
using namespace std;
const long long N=100010;
long long n,a[N],m,p1,p2,s1,s2,longg,hu;
int main() {
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i];
	cin>>m>>p1>>s1>>s2;
	a[p1]+=s1;
	long long ans=m,minn;
	for(int i=1;i<=m;i++) longg+=(m-i)*a[i];
	for(int i=m+1;i<=n;i++) hu+=(i-m)*a[i];
	minn=labs(longg-hu);
	for(int i=1;i<m;i++){
		longg+=(m-i)*s2;
		if(abs(longg-hu)<minn){
			ans=i;
			minn=abs(longg-hu);
		}
		longg-=(m-i)*s2;
	}
	for(int i=m+1;i<=n;i++){
		hu+=(i-m)*s2;
		if(abs(longg-hu)<minn){
			ans=i;
			minn=abs(longg-hu);
		}
		hu-=(i-m)*s2;
	}
	cout<<ans;
	return 0;
}