打卡信奥刷题（3403）用C++实现信奥题 P10051 [CCO 2022] Rainy Markets

P10051 [CCO 2022] Rainy Markets

题目背景

由于数据包过大，本题无法上传全部数据。

题目描述

有 $N$ 个公交车站，标号为 $1,\dots ,N$。第 $i$ 个公交车站可以容纳 $B_i$ 个人。

对于每个 i∈{1,…,N−1}i \in\{1, \ldots, N-1\}i∈{1,…,N−1}，有一条人行道连接公交车站 $i$ 和公交车站 $i+1$，中间有一个露天市场。第 $i$ 个市场有 $U_i$ 把雨伞出售，每把雨伞的价格为 $1$。

现在，有 $P_i$ 个人在第 $i$ 个市场里面，所有的公交车站都是空的。

突然，天开始下雨，市场 $i$ 的每个人都必须在三种方案中选择一种：

• 去公交车站 $i$；
• 去公交车站 $i+1$；
• 留下来买一把雨伞。

如果一个人无法在某个公交车站下或者买一把雨伞，他们就会淋湿。

你需要回答如果在最优的安排方案下，能否确保每个人都能不被雨淋湿。如果是的话，你需要给出他们需要花费的最少的钱，以及每个人应该移动到哪个公交车站。

输入格式

第一行包含一个整数 $N$。

第二行包含 $N$ 个用空格分隔的整数 $B_i(1\le i\le N)$，表示公交车站 $i$ 的容量。

第三行包含 $N-1$ 个用空格分隔的整数 $P_i(1\le i\le N-1)$，表示市场 $i$ 的人数。

第四行包含 $N-1$ 个用空格分隔的整数 $U_i(1\le i\le N-1)$，表示市场 $i$ 出售的雨伞的数量。

输出格式

如果每个人都能在雨伞或公交车站下，输出 $N+1$ 行：

• 第一行输出一行 YES。
• 第二行输出一个整数，表示包含在雨伞上花费的最少的钱。
• 接下来的 $N-1$ 行，每行输出三个用空格分隔的整数分别表示市场 $i(1\le i\le N-1)$ 移动到公交车站 $i$ 的人数，市场 $i$ 买雨伞的人数，市场 $i$ 移动到公交车站 $i+1$ 的人数。

如果有多种合法方案，你可以输出任意一种。

否则，输出一行 NO。

输入输出样例 #1

输入 #1

3
10 15 10
20 20
0 0

输出 #1

NO

输入输出样例 #2

输入 #2

3
10 15 10
20 20
0 11

输出 #2

YES
5
10 0 10
5 5 10

说明/提示

样例 1 解释

公交车站有 $35$ 个空位，没有雨伞出售，但市场有 $40$ 个人，所以答案是 NO。

样例 2 解释

市场 $1$ 中的 $10$ 个人会去公交车站 $1$，没有人会买雨伞，$10$ 个人会去公交车站 $2$。

市场 $2$ 中的 $5$ 个人会去公交车站 $2$，$5$ 个人会留下来买雨伞，$10$ 个人会移动到公交车站 $3$。

总共购买了 $5$ 把雨伞，花费了 $5$。

数据范围

对于所有的数据，有 $2\le N\le 10^6$，$0\le B_i\le 2\cdot 10^9$，$0\le P_i,U_i\le 10^9$。

C++实现

#include<iostream>
using namespace std;
#define int long long 
const int N=1e6+10;
int n;
int b[N],p[N],u[N];
int f[N];
int l[N],r[N],buy[N];
signed main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		scanf("%lld",&b[i]);
	}
	for(int i=1;i<n;i++){
		scanf("%lld",&p[i]);
	}
	for(int i=1;i<n;i++){
		scanf("%lld",&u[i]);
		if(u[i]+b[i]+b[i+1]<p[i]){
			cout<<"NO";
			return 0;
		}
	}
	for(int i=1;i<=n;i++){
		f[i]=max(0ll,b[i]-max(p[i-1]-f[i-1],0ll));
	}
	long long res=0;
	for(int i=n-1;i;i--){
		r[i]=min(p[i],b[i+1]-l[i+1]);
		p[i]-=r[i];
		l[i]=min(p[i],f[i]);
		p[i]-=l[i];
		buy[i]=min(p[i],u[i]);
		p[i]-=buy[i];
		l[i]+=p[i];
		res+=buy[i];
		if(l[i]>b[i]){
			cout<<"NO";
			return 0;
		}
	}
	printf("YES\n%lld\n",res);
	for(int i=1;i<n;i++){
		printf("%lld %lld %lld\n",l[i],buy[i],r[i]);
	}
}

后续

接下来我会不断用C++来实现信奥比赛中的算法题、GESP考级编程题实现、白名单赛事考题实现，记录日常的编程生活、比赛心得，感兴趣的请关注，我后续将继续分享相关内容