「一本通 1.1 练习 4」家庭作业 -- 贪心+并查集

loj 10008

题目分析：

• 将按照$w从大到小排序，放的越后越好$
• 维护一个$f[i]表示i之前能放的最靠后的位置，用并查集实现$

Code:

#include <bits/stdc++.h>
using namespace std;
#define maxn 1000010
#define maxd 700010

int n,f[maxd],m=0;
struct node {
	int d,w;
}e[maxn];

inline int read_() {
	int x=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9') {
		if(c=='-') f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9') {
		x=(x<<1)+(x<<3)+c-'0';
		c=getchar();
	}
	return x*f;
}

inline bool cmp_(node aa,node bb) {
	return aa.w > bb.w;
}

int find_(int x) {
	if(f[x]==x) return x;
	else return f[x]=find_(f[x]);
}

void readda_() {
	n=read_();
	f[0]=0;
	for(int i=1;i<=n;++i) {
		e[i].d=read_();e[i].w=read_();
		m=max(m,e[i].d);
	}
	for(int i=0;i<=m;++i) f[i]=i;
	sort(e+1,e+n+1,cmp_);
	long long ans=0;
	for(int i=1;i<=n;++i) {
		int pdc=find_(e[i].d);
		if(pdc!=0) {
			f[pdc]=pdc-1;
			ans+=e[i].w;
		}
	}
    printf("%lld",ans);
}

int main() {
	freopen("a.txt","r",stdin);
	readda_();
	return 0;
}