P5016 [NOIP 2018 普及组] 龙虎斗

原创于 2025-08-09 23:05:55 发布 · 900 阅读

10 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

题目背景

NOIP2018 普及组 T2

题目描述

轩轩和凯凯正在玩一款叫《龙虎斗》的游戏，游戏的棋盘是一条线段，线段上有 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，表示你选择的兵营编号。如果存在多个编号同时满足最优，取最小的编号。

输入输出样例

输入 #1复制

6 
2 3 2 3 2 3 
4 6 5 2

输出 #1复制

输入 #2复制

6 
1 1 1 1 1 16 
5 4 1 1

输出 #2复制

说明/提示

样例 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≤10^5,ci,s1,s2≤10^9。

//分析//

本题中n的数据规模为10的5次方，如果用枚举来做时间复杂度并不是很大,所以本文章的做法是枚举。

首先，我们要知道在神兵天降之后，龙与虎的气势差值是多少，这样才能求出在哪个兵营放兵才能使差值最小。通过读题可知如何计算气势，只需求出它们相减的绝对值，就知道它们的差值了。接下来就要进行枚举了。从1到n的所有兵营都要看，而它又分三种情况。一是看的兵营是m号，这种情况不能忘记写。因为说不定就算天降神兵，龙与虎的差值也很小，这时候再加上新兵反而会拉开差距，所以这时候把新兵放到m号才是最好的选择。另外两种情况自然是把新兵放到龙阵营或虎阵营。每看完一个，要把这时的差值和最小差值做比较，如果更小就更新，反之不变。最后输出最小差值所对应的兵营编号。

//代码//

#include <bits/stdc++.h>        //万能头文件
using namespace std;
long long n,c[100005],m,p1,s1,s2,minn=1e9,ans,p2;
                                //兵营总数n，每一个兵营中工兵的数量c数组，分界线m号兵营的位置m，天降神兵的位置p1，天降神兵的数量s1，可以放的新兵数量s2，最小差值minn，放下新兵后的差值ans，最小差值所对应的兵营编号p2
long long l,h,cj;               //龙阵营的气势l，虎阵营的气势h，天降神兵后的差距cj
int main() {
	cin>>n;                     //输入兵营总数
	for (int i=1;i<=n;i++) {    //循环输入兵营中的工兵数
		cin>>c[i];              //存入c数组
	}
	cin>>m>>p1>>s1>>s2;         //输入m号兵营的位置，天降神兵的位置与数量，可以放的新兵数量
	c[p1]+=s1;                  //直接将天降神兵放到对应的兵营里
	for (int i=1;i<m;i++) {     //m号兵营的左边是龙阵营
		l+=c[i]*(m-i);          //计算龙阵营的气势
	}
	for (int i=m+1;i<=n;i++) {  //m号兵营的右边是虎阵营
		h+=c[i]*abs(m-i);       //计算虎阵营的气势
    }
	cj=abs(l-h);                //计算差距
	for (int i=1;i<=n;i++) {    //从第一个开始看，一直到最后一个
		if (i==m) {             //如果当前看的兵营是m号兵营
			ans=cj;             //差距不变
		}
		else if (i<m) {         //如果当前看的兵营是龙阵营的
			ans=abs((l+((c[i]+s2)*(m-i)-c[i]*(m-i)))-h);
                                //差距就等于龙阵营原先的气势加上新增加的气势减去虎阵营的气势
		}
		else if (i>m) {         //如果当前看的兵营是虎阵营的
			ans=abs((h+((c[i]+s2)*(i-m)-c[i]*(i-m)))-l);
                                //差距就等于虎阵营原先的气势加上新增加的气势减去龙阵营的气势
		}
		if (ans<=minn) {        //只要在i号兵营增加新兵后的差距小于最小差值
			minn=ans;           //更新最小差值
			p2=i;               //让最小差值所对应的兵营编号也更新
		}
	}
	cout<<p2;                   //输出最小差值所对应的兵营编号
	return 0;
}

标签

#模拟 #枚举 #c++