CHDOJ——P2124采购礼品

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

CHDOJ——P2124采购礼品

思路

集合分类+DP
并查集将同一个套餐的礼品的价钱和价值加在一起看作一个整体,并将这些集合放在一个 v e c t o r vector vector容器当中,然后我们通过 01 背 包 01背包 01背包的思路来处理。
也就是如下的公式。

d p [ j ] = m a x ( d p [ j ] , d p [ j − v [ i ] ] + w [ i ] ) dp[j] = max(dp[j],dp[j-v[i]]+w[i]) 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;
}

加油!冲冲冲!!!

上一篇:练习8-2 计算两数的和与差 (10 分)


下一篇:YbtOJ-序列计数【组合数学,莫队】