[NOIP2018 普及组] 龙虎斗

题目背景

        NOIP2018 普及组 T2

题目描述

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

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

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

输入格式

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

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

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

cin>>n;
for(i=1;i<=n;i++)
{
	cin>>c[i];
}
cin>>m>>p>>s1>>s2;

输出格式

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

if(l==h)
{
	cout<<m;
	return 0;
}

cout<<minn;

输入输出样例

输入 #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 说明

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

双方以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 ≤ p 1​ ≤ 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。

AC代码

#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;	
}