【洛谷P1230】智力大冲浪

题目大意：给定 N 项任务，每项任务有一个截至完成时间，若在截止时间之后完成要罚款 \(w_i\) 元，最初有 M 元，怎样完成能够留下最多得钱。

题解：按照罚款从多到少贪心，在查找能够最晚完成一项任务的时间时，可以采用并查集优化，即：建立一个时间上的并查集，每次在某个时间完成一项任务时，合并当前时间和前一个时间，这样每次查找时省去了一个循环的时间。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=510;

int n,m,f[maxn];
struct node{int t,w;}e[maxn];
bool vis[maxn];

int find(int x){
	return x==f[x]?x:f[x]=find(f[x]);
}

bool cmp(const node& x,const node& y){
	return x.w>y.w;
}

void read_and_parse(){
	scanf("%d%d",&m,&n);
	for(int i=1;i<=n;i++)scanf("%d",&e[i].t);
	for(int i=1;i<=n;i++)scanf("%d",&e[i].w),f[i]=i;
	sort(e+1,e+n+1,cmp);
}

void solve(){
	for(int i=1;i<=n;i++){
		int fa=find(e[i].t);
		if(!fa)m-=e[i].w;
		else f[fa]=fa-1;
	}
	printf("%d\n",m);
}

int main(){
	read_and_parse();
	solve();
	return 0;
}