信息学奥赛一本通 1979：【18NOIP普及组】龙虎斗

【题目描述】

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

下面图1为n=6�=6的示例：

图 1. n=6的示例

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

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

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

图 2. n=6,m=4�=6,�=4的示例

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

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

【输入】

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

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

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

【输出】

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

【输入样例】

6
2 3 2 3 2 3
4 6 5 2

【输出样例】

2

代码：

#include<iostream>
#include<cmath> 
using namespace std;
int main(){
long long a[100001],n,m,p1,s1,s2;
cin >> n;
for(int i = 1; i <= n; i++){
cin >> a[i];
}
cin >> m >> p1 >> s1 >> s2;
a[p1] += s1;
long long lpower = 0,rpower = 0;
for(int i = 1; i < m; i++){//龙气势 
lpower += a[i] * (m - i);
}
for(int i = m + 1; i <= n; i++){//虎气势 
rpower += a[i] * (i - m);
}
long long grap = abs(lpower - rpower);//龙虎气势差 
long long p2 = m;//默认 
for(int i = 1; i <= n; i++){
if(i < m){
if(abs(lpower + s2 * (m - i) - rpower) < grap){
grap = abs(lpower + s2 * (m - i) - rpower);
p2 = i;
}
}else if(i > m){
if(abs(rpower + s2 * (i - m) - lpower) < grap){
grap = abs(rpower + s2 * (i - m) - lpower);
p2 = i;
}
}
}
cout << p2;


return 0;
}