题目描述:
轩轩和凯凯正在玩一款叫《龙虎斗》的游戏,游戏的棋盘是一条线段,线段上有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,表示你选择的兵营编号。如果存在多个编号同时满足最优,取最小的编号。
限制:
空间限制:128MByte
时间限制:1秒
样例:
输入:
输入 #1
6
2 3 2 3 2 3
4 6 5 2
输入 #2
6
1 1 1 1 1 16
5 4 1 1
输出:
输出 #1
2
输出 #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 ≤ 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≤109
注意数据范围!(我被这个坑到哭TAT)一定要用 long long类型
————————————————————华丽的分割线—————————————————
不多说了,一切尽在代码掌控之中:
#include<iostream>
#include<algorithm>
using namespace std;
int n;
long long t,m,p1,s1,s2,sum1=0,sum2=0;
long long a[11111111];
long long Min=0x3f3f3f3f;
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]);
}
scanf("%lld%lld%lld%lld",&m,&p1,&s1,&s2);
a[p1]+=s1;
for(int i=1;i<=m;i++){
sum1+=a[i]*(m-i);
}
for(int i=m;i<=n;i++){
sum2+=a[i]*(i-m);
}
t=m;
for(int i=1;i<=n;i++){
long long t1=sum1,t2=sum2;
if(i<m){
t1+=s2*(m-i);
} else if(i>m){
t2+=s2*(i-m);
}
if(Min>abs(t1-t2)){
Min=abs(t1-t2);
t=i;
}
}
printf("%lld\n",t);
return 0;
}
//AC了 TAT
|