洛谷 AT_abc358_d [ABC358D] Souvenirs 题解

题目翻译

AtCoder 乐园出售 $N$ 种纪念品盒子，标号依次为 $1\sim N$，盒子 $i$ 的价格为 $A_i$ 元，有 $A_i$ 个糖果。

高桥要从中买 $M$ 个盒子，然后分别给每个人各一个盒子，第 $i$ 个人的盒子中至少要有 $B_i$ 个糖果。

请问高桥是否能买到符合要求的 $M$ 个盒子，如果能就输出高桥所需要的最小金额，如果不能就输出 -1。

题目分析

为了让金额最小，所以给第 $i$ 个人买的盒子中的糖果数量得是大于等于 $B_i$ 的数中，最小的一个。如果 $A$ 和 $B$ 都是升序序列，再设 $A$ 中大于等于 $B_i$ 的数中的最小的一个是 $A_j$，那么 $A$ 中大于等于 $B_{i+1}$ 的数中最小的一个数的下标绝对大于 $j$。所以可以先将 $A$ 和 $B$ 进行升序排序，然后一个个寻找。需要注意的是，再续找的过程中还要判断是否有可行的方案。

Code

#include<iostream>
#include<algorithm>
#define int long long
using namespace std;
signed main() {
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	int n,m,a[200000],b[200000],ans=0;
	cin>>n>>m;
	for(int i=0;i<n;++i) cin>>a[i];
	for(int i=0;i<m;++i) cin>>b[i];
	sort(a,a+n);
	sort(b,b+m);
	for(int i=0;i<m;++i)
		for(static int j=0;j<n;++j) {
			if(n-j<m-i||(a[j]<b[i]&&n-j==m-i)) {//如果剩余的盒子数量不够就输出-1
				cout<<-1;
				return 0;
			}
			if(a[j]>=b[i]) {
				ans+=a[j];//累加最小金额
				++j;//这里还要将下标加1
				break;
			}
		}
	cout<<ans;
	return 0;
}