Vijos P1250最勇敢的机器人 dp 分组背包

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N =1010;
struct node{
	int val;
	int cost;
}s[N];
int set[N];
int dp[N];
vector<int>g[N];
bool flag[N];
int temp[N];
int find(int x){
	int r=x;
	while(r!=set[r])
		r=set[r];
	int temp;
	while(set[x]!=r){
		temp=set[x];
		set[x]=r;
		x=temp;
	}
	return r;
}
void insert(int x,int y){
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy) set[fx]=fy;
}
int main(){
	int n,m,k;
	while(scanf("%d%d%d",&n,&m,&k)!=EOF){
		for(int i=1;i<=n;i++){
			scanf("%d%d",&s[i].val,&s[i].cost);
			set[i]=i;
			g[i].clear();
		}
		int r,c;	
		while(k--){
			scanf("%d%d",&r,&c);
			insert(r,c);
		}
		memset(flag,false,sizeof(flag));
		int cnt=0;
		for(int i=1;i<=n;i++){
			int fx=find(set[i]);
			g[fx].push_back(i);
			if(flag[fx]==false) temp[cnt++]=fx;
			flag[fx]=true;
		}
		memset(dp,0,sizeof(dp));
		for(int i=0;i<cnt;i++){
			for(int j=m;j>=0;j--){
				for(int h=0;h<g[temp[i]].size();h++){
					int u=g[temp[i]][h];
					if(s[u].cost<=j){
						dp[j]=max(dp[j],dp[j-s[u].cost]+s[u].val);
					}
				}
			}
		}
		printf("%d\n",dp[m]);
	}
	return 0;
}

用并查集进行分组，然后分组背包。