题目背景
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;
}