[数据结构与算法]CHDOJ——P2124采购礼品

这是一道并查集和01背包问题的结合，题目难度不大，刚开始接触并查集的可以试试这一道
这是链接https://oj.chdacm.cn/problem.php?id=2124

思路

集合分类+DP
并查集将同一个套餐的礼品的价钱和价值加在一起看作一个整体，并将这些集合放在一个$vector$容器当中，然后我们通过$01背包$的思路来处理。
也就是如下的公式。
$dp[j]=max(dp[j],dp[j?v[i]]+w[i])$
核心代码如下：

dp[j] = max(dp[j],dp[j-num[sum[i]].first]+num[sum[i]].second);

代码

下面是AC代码，以供参考

#include<iostream>
#include<algorithm>
#include<iomanip>
#include<cstring>
#include<vector>
#include<cmath>
#define ll long long
#define mem(s,i) memset(s,i,sizeof(s))
#define PII pair<int, int>
#define INF 0x7fffffff
using namespace std;
const int N = 1e4+10;
int n, m, k;
int pre[N];
int dp[N];
PII vis[N];
PII num[N];
vector<int> sum;
int find(int x){
    if(pre[x]!=x)pre[x] = find(pre[x]);
    return pre[x];
}
void merge(int x, int y){
    pre[find(x)] = find(y);
}
void solve(){
    cin>>n>>m>>k;
    for(int i = 1;i <= n;i++)pre[i] = i;
    for(int i = 1;i <= n;i++)cin>>vis[i].first>>vis[i].second;
    for(int i = 0;i < m;i++){
        int x, y;
        cin>>x>>y;
        merge(x,y);
    }
    for(int i = 1;i <= n;i++){
        int res = find(i);
        auto ans = find(sum.begin(),sum.end(),res);
        if(ans==sum.end())
            sum.push_back(res);
        num[res].first += vis[i].first;
        num[res].second += vis[i].second;
    }
    for(int i = 0;i < sum.size();i++){
        for(int j = k;j >= num[sum[i]].first;j--){
            dp[j] = max(dp[j],dp[j-num[sum[i]].first]+num[sum[i]].second);
        }
    }
    // for(int i = 0;i < sum.size();i++)cout<<dp[i]<<" ";cout<<endl;
    cout<<dp[k]<<endl;
}
int main(){
    solve();
    return 0;
}

加油！冲冲冲！！！