【YbtOJ算法高效进阶2023 「图论」第1章 并查集 #D】超市购物(并查集)

题目大意

一个商店有很多批货，每一批货又分别有 $n$ 个商品，同时每一样商品都有收益 $P_i$，和过期时间 $D_i$，每天只能卖出一件，一旦超过了过期时间，商品就不能再卖。你要做的就是求出每批货最多能得到多少收益。

对于 $100\%$ 的数据 ，$1\le n\le 10^4$，$1\le D_i,P_i\le 10^4$。

输入格式
多组数据。
每组先给出一个整数 $n$，表示这批货的商品个数。
然后有 $n$ 对数，每对数有两个用空格隔开的正整数 $P_i,D_i$，表示第 $i$个商品的收益和过期时间。相邻两对数之间用空格隔开。输入以一个文件终止符结束，并且保证所有输入数据正确。

输出格式
对于每组数据，输出一行一个整数表示该批货物能卖出的最大价格。

输入样例

4 50 2 10 1 20 2 30 1
7 20 1 2 1 10 3 100 2 8 2 5 20 50 10

输出样例

80 
185

基本思路

这题乍一看和并查集没啥关系。

一个十分明显的贪心。我们先按收益给它排个序，优先售出收益高的商品，对于每一个商品，我们在它的最后期限售出，最好是在临过期时售出。如果临过期那天被前面收益更大的商品占用了呢？这个好办，往前倒到最后一个没被占用的日期就行了啊。

可问题是你如此查找的话复杂度是$O(n^2)$，明显承受不了。怎么办呢？这个时候就要祭出我们的并查集了。

怎么用呢？我们的困境是如何跳转至前面最后一个可用日期。论查找根节点并查集效率十分高。于是我们把日期抽象成一个集合，每个集合的代表元素就是前面最后一个可用节点，其它都是后面已经被占用的日期，这样如果我们发现临过期的日期被占用了，我们就顺着并查集找到它的根节点，根节点就是可用日期，将根节点标记占用，然后将根节点并入前一个日期所在集合，这样下次还是能找到可用日期。

如若发现根节点为0，则代表真的没有日期可用了。

核心代码

（你别说在YbtOJ上格式化的代码就是好看）

（在这里还是原始代码好看一些）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e4+10;
struct node{
	int pi,di;
}a[N];
int n,fa[N],ans,dmax;
//fa[i]表示日期为i的父节点
inline int find(int x){
	if(fa[x]==x) return x;
	return fa[x]=find(fa[x]);
}
inline void merge(int x,int y){
	int xx=find(x),yy=find(y);
	if(xx!=yy) fa[xx]=yy;
}
inline void init(){
	for(int i=0;i<=dmax;i++) fa[i]=i;
}
bool cmp(node a,node b){
	return a.pi>b.pi;
}
int main(){
	while(cin>>n){//当还没读完 
		dmax=ans=0;
		for(int i=1;i<=n;i++){
			scanf("%d%d",&a[i].pi,&a[i].di);
			dmax=max(dmax,a[i].di);
		}
		init();
		sort(a+1,a+1+n,cmp);
		for(int i=1;i<=n;i++){
			int cnt=find(a[i].di);
			if(cnt==0) continue;
			//前面日期都用完了,但不是所有 
			ans+=a[i].pi;
			merge(cnt,cnt-1);
			//把此日期并入前一天的集合里
			//一个集合只能是代表元素没使用过,这样
			//才方便查找 
		}
		printf("%d\n",ans);
	}
	return 0;
}