K11755 龙虎斗

题目描述

轩轩和凯凯正在玩一款叫《龙虎斗》的游戏,游戏的棋盘是一条线段,线段上有n个兵营(自左至右编号 1~n),相邻编号的兵营之间相隔1厘米,即棋盘为长度为n−1 厘米的线段。i号兵营里有ci位工兵。下图1为n=6的示例:

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

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

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

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

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

输入格式

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

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

接下来的一行包含四个正整数,相邻两数间以一个空格分隔,分别代表m,p1,s1,s2。

输出格式

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

输入输出样例

输入样例1：

6 
2 3 2 3 2 3 
4 6 5 2

输出样例1：

2

输入样例2：

6 
1 1 1 1 1 16 
5 4 1 1

输出样例2：

1

说明

【样例1说明】

双方以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

此时双方气势相等。

【样例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

当你将手中的s2=1位工兵派往p2=1号兵营时,龙方的气势变为:

11 + 1 × (5 − 1) = 15

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

【数据规模与约定】

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 ≤ 10^5, ci, s1, s2 ≤ 10^9。

【耗时限制】1000ms 【内存限制】256MB

//
//Created by Carlgood.
//
//Note:This program is written in version DEV-C++ 5.11.
# include<iostream>
# include<cmath>
# include<string>
# include<cstring>
# include<cstdio>
# include<algorithm>
# include<sstream>
# include<iomanip>
# include<vector>
# include<queue>
# include<deque>
# include<stack>
# include<stdio.h>
# define This_program_is_written_by_Carlgood_Programming_Studio 9876543210
using namespace std;
long long n,m,p1,s1,s2,a[100010],ans;
long long l,r,minn=0x7fffffffffffffff;
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	cin>>m>>p1>>s1>>s2;
	a[p1]+=s1;
	for(int i=1;i<=m-1;i++) l+=a[i]*(m-i);
	for(int i=m+1;i<=n;i++) r+=a[i]*(i-m);
	for(int i=1;i<=n;i++)
	{
		long long l2=l,r2=r;
		if(i<m) l2+=s2*(m-i);
		if(i>m) r2+=s2*(i-m);
		long long dis=labs(l2-r2);
		if(dis<minn)
		{
			minn=dis;
			ans=i;
		}
	}
	cout<<ans;
	return 0;
}