题意分析
首先用并查集处理所有的分组,将不同分组放入不同vector中。
然后做分组背包即可,要注意分组背包的顺序
for 所有分组 : 0 to g
for 背包容量 : v to 0 // 必须逆序,如果写反,对于同一件物品可以无限装
for 分组物品 : 0 to item
dp[j] = max(dp[j], dp[j - w[item]] + v[item])
代码总览
#include<bits/stdc++.h>
using namespace std;
const int nmax = 1005;
int fa[nmax];
int v[nmax],w[nmax];
int dp[nmax];
int n,p,m,g;
vector<int> ve[nmax];
map<int,int> mp;
int findset(int x){
int rt = x,temp;
while(fa[rt] != rt) rt = fa[rt];
while(x != rt){
temp = fa[x];
fa[x] = rt;
x = temp;
}
return rt;
}
void unionset(int x, int y){
x = findset(x);
y = findset(y);
if(x == y) return;
else fa[x] = y;
}
void getdp(){
for(int i = 0;i<g;++i){
for(int j = p;j>=0;--j){
for(int k = 0;k<ve[i].size();++k){
int t = ve[i][k];
if(j>=w[t]) dp[j] = max(dp[j],dp[j - w[t]] + v[t]);
}
}
}
}
void debug(){
for(int i = 0;i<g;++i){
for(int j = 0;j<ve[i].size();++j){
printf("%d ",ve[i][j]);
}
printf("\n");
}
}
int main(){
scanf("%d %d %d",&n,&p,&m);
for(int i = 0;i<=n;++i) fa[i] = i;
for(int i = 1;i<=n;++i) scanf("%d %d",&v[i],&w[i]);
int u,v;
for(int i = 0;i<m;++i){
scanf("%d %d",&u,&v);
unionset(u,v);
}
for(int i = 1;i<=n;++i){
int t = findset(i);
if(mp.count(t) == 0) mp[t] = g++;
ve[mp[t]].push_back(i);
}
// debug();
getdp();
printf("%d\n",dp[p]);
return 0;
}