求解
轩轩和凯凯正在玩一款叫《龙虎斗》的游戏,游戏的棋盘是一条线段,线段上有 nn 个兵营(自左至右编号 1 ∼n),相邻编号的兵营之间相隔 1 厘米,即棋盘为长度为 n−1 厘米的线段。ii号兵营里有 ci
位工兵。 下面图 1 为 n=6 的示例:
轩轩在左侧,代表“龙”;凯凯在右侧,代表“虎”。 他们以 mm 号兵营作为分界, 靠左的工兵属于龙势力,靠右的工兵属于虎势力,而第 mm 号兵营中的工兵很纠结,他们不属于任何一方。
一个兵营的气势为:该兵营中的工兵数\times× 该兵营到 mm 号兵营的距离;参与游戏 一方的势力定义为:属于这一方所有兵营的气势之和。
下面图 2 为 n = 6,m = 4的示例,其中红色为龙方,黄色为虎方:
s_1位工兵突然出现在了 pp1号兵营。作为轩轩和凯凯的朋友,你知道如果龙虎双方气势差距太悬殊,轩轩和凯凯就不愿意继续玩下去了。为了让游戏继续,你需要选择一个兵营 p_2,并将你手里的 s2
位工兵全部派往 兵营 p2,使得双方气势差距尽可能小。
注意:你手中的工兵落在哪个兵营,就和该兵营中其他工兵有相同的势力归属(如果落在 m号兵营,则不属于任何势力)。
输入
输入文件的第一行包含一个正整数n,代表兵营的数量。
接下来的一行包含 n 个正整数,相邻两数之间以一个空格分隔,第 i个正整数代 表编号为 ii 的兵营中起始时的工兵数量 ci
。
接下来的一行包含四个正整数,相邻两数间以一个空格分隔,分别代表 m,p1,s1,s2。
输出
输出文件有一行,包含一个正整数,即 p2,表示你选择的兵营编号。如果存在多个编号同时满足最优,取最小的编号。
样例输入
6
2 3 2 3 2 3
4 6 5 2
样例输出
2
提示
【数据规模与约定】
1 <m <n, 1 ≤ p1≤n。
对于20%的数据,n=3,m=2, ci= 1,s1, s2≤100。
另有20%的数据,n≤10, 1= 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。
下面是我的代码,能通过样例,但是过不了#include<stdio.h>
#include<math.h>
int main()
{
int n,i,m,p1;
long long int sum1=0,sum2=0,l,l1,l2,abc1,abc2,s1,s2;
long long int a[100001];
scanf("%lld",&n);
for(i=0;i<n;i++)
{
scanf("%lld ",&a[i]);
}
scanf("%d %d %lld %lld",&m,&p1,&s1,&s2);
for(i=0;i<m;i++)
{
sum1+=a[i]*(m-1-i);
}
for(i=m;i<n;i++)
{
sum2+=a[i]*(i-m+1);
}
if(p1>m)
sum2+=s1*(p1-m);
if(p1<m)
sum1+=s1*(m-p1);
l=abs(sum2-sum1);
l1=l/s2;
l2=l1+1;
if(sum1>sum2)
{
if(l1+m>n)
l1=n-m-1;
if(l2+m>n)
l2=n-m-1;
abc1=abs(sum2+l1*s2-sum1);
abc2=abs(sum2+l2*s2-sum1);
if(abc1<=abc2)
printf("%d",m+l1);
else
printf("%d",m+l2);
}
else
if(l1>m)
l1=m-1;
if(l2>m)
l2=m-1;
abc1=abs(sum1+l1*s2-sum2);
abc2=abs(sum1+l2*s2-sum2);
if(abc2<=abc1)
printf("%d",m-l2);
else
printf("%d",m-l1);
return 0;
}
可能数据太大时溢出了,数组a开大一些