龙虎斗(fight)

题目描述

【问题描述】

轩轩和凯凯正在玩一款叫《龙虎斗》的游戏， 游戏的棋盘是一条线段， 线段上有 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 ，表示你选择的兵营编号。如果存在多 个编号同时满足最优，取最小的编号。

【输入输出样例 1】

 

fight			in		fight	out
				6 2 3 2 3 2 3 4 6 5 2	2

 

 

 

见选手目录下的 fight/fight1.in 和 fight/fight1.ans。

【输入输出样例 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

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

此时双方气势相等。

【输入输出样例 2】

 

fight			in		fight	out
				6 1 1 1 1 1 16 5 4 1 1	1

见选手目录下的 fight/fight2.in 和 fight/fight2.ans。

【输入输出样例 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

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

 

【输入输出样例 3】

见选手目录下的 fight/fight3.in 和 fight/fight3.ans。

 

【数据规模与约定】

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

代码

#include<bits/stdc++.h>
using namespace std;
long long c[100005];
int main()
{ 
	long long i,n,h=0,l=0,m,p,s1,s2,t=0,mini=1000000000,minn=0;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		cin>>c[i];
	}
	cin>>m>>p>>s1>>s2;
	c[p]+=s1;
	for(i=1;i<=m;i++)
	{
		l+=c[i]*(m-i);
	}
	for(i=n;i>=m;i--)
	{
		h+=c[i]*(i-m);
	}
	if(l==h)
	{
		cout<<m;
		return 0;
	}
	else if(l<h)
	{
		for(i=1;i<=m;i++)
		{
			t=l+s2*(m-i);
			if(abs(t-h)<mini) 
			{
				mini=abs(t-h);
				minn=i;
			}
		}
	}
	else 
	{
		for(i=n;i>=m;i--)
		{
			t=h+s2*(i-m);
			if(abs(t-l)<mini) 
			{
				mini=abs(t-l);
				minn=i;
			}
		}
	}
	cout<<minn; 
	return 0;	
}