轩轩和开开正在玩一款叫《龙虎斗》的游戏,游戏的棋盘是一条线段,线段上有n个兵营(自左至右编号1~n),相邻编号的兵营之间相隔1厘米,即棋盘为长度为n-1厘米的线段。i号兵营里有ci位工兵。
下面图1为n = 6的示例:
轩轩在左侧,代表“龙”;凯凯在右侧,代表“虎”。 他们以 m 号兵营作为分界, 靠左的工兵属于龙势力,靠右的工兵属于虎势力,而第 𝐦 号兵营中的工兵很纠结,他 们不属于任何一方。
一个兵营的气势为:该兵营中的工兵数 × 该兵营到 m 号兵营的距离;参与游戏 一方的势力定义为:属于这一方所有兵营的气势之和。
下面图 2 为 n = 6, m = 4 的示例,其中红色为龙方,黄色为虎方:
游戏过程中,某一刻天降神兵,共有 s1 位工兵突然出现在了p1号兵营。作为轩轩和凯凯的朋友,你知道如果龙虎双方气势差距太悬殊,轩轩和凯凯就不愿意继续玩下去了。为了让游戏继续,你需要选择一个兵营 p2,并将你手里的 s2位工兵全部派往 兵营p2,使得双方气势差距尽可能小。
注意:你手中的工兵落在哪个兵营,就和该兵营中其他工兵有相同的势力归属(如果落在 m 号兵营,则不属于任何势力)。
输入描述:
输入文件的第一行包含一个正整数 n,代表兵营的数量。
接下来的一行包含 n 个正整数,相邻两数之间以一个空格分隔,第 i 个正整数代表编号为 i 的兵营中起始时的工兵数量 ci。接下来的一行包含四个正整数,相邻两数间以一个空格分隔,分别代表 m, p1, s1, s2。
输出描述:
输出文件有一行,包含一个正整数,即 p2,表示你选择的兵营编号。如果存在多个编号同时满足最优,取最小的编号。
输入:
6
2 3 2 3 2 3
4 6 5 2
输出:
2
备注:
1<m<n1<m<n,
1≤p1≤n1≤p1≤n,
n≤10^5,ci,s1,s2≤10^9
解析:直接暴力枚举每一个军营加入s2,然后更新答案即可,需要开long long。
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+5;
typedef long long ll;
ll a[N];
void solve()
{
int n,m,p1;
ll s1,s2,sum1=0,sum2=0;//sum1,sum2分别记录两个军营的势力
scanf("%d",&n);
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
scanf("%d%d%lld%lld",&m,&p1,&s1,&s2);
a[p1]+=s1;//s1个加入p1军营
for(int i=1;i<m;i++) sum1+=a[i]*(m-i);//左边军营势力
for(int i=m+1;i<=n;i++) sum2+=a[i]*(i-m);//右边军营势力
int ans=m;//记录下答案位置,可以先在m点(中立)
ll minn=abs(sum1-sum2);//记录最小差距
for(int i=1;i<m;i++)//加入左边军营
{
ll k=sum1+s2*(m-i);
if(abs(k-sum2)<minn) minn=abs(k-sum2),ans=i;
}
for(int i=m+1;i<=n;i++)//加入右边军营
{
ll k=sum2+s2*(i-m);
if(abs(k-sum1)<minn) minn=abs(k-sum1),ans=i;
}
printf("%d\n",ans);
}
int main()
{
solve();
return 0;
}