算法提高课整理
CSDN个人主页:更好的阅读体验
原题链接
题目描述
Joe觉得云朵很美,决定去山上的商店买一些云朵。
商店里有 n 朵云,云朵被编号为 1,2,…,n,并且每朵云都有一个价值。
但是商店老板跟他说,一些云朵要搭配来买才好,所以买一朵云则与这朵云有搭配的云都要买。
但是Joe的钱有限,所以他希望买的价值越多越好。
输入格式
第 1 行包含三个整数 n,m,w,表示有 n 朵云,m 个搭配,Joe有 w 的钱。
第 2∼n+1行,每行两个整数 ci,di 表示 i 朵云的价钱和价值。
第 n+2∼n+1+m 行,每行两个整数 ui,vi,表示买 ui 就必须买 vi,同理,如果买 vi 就必须买 ui。
输出格式
一行,表示可以获得的最大价值。
数据范围
1≤n≤10000,
0≤m≤5000,
1≤w≤10000,
1≤ci≤5000,
1≤di≤100,
1≤ui,vi≤n
输入样例:
5 3 10
3 10
3 10
3 10
5 100
10 1
1 3
3 2
4 2
输出样例:
1
思路
一个比较板子的有依赖背包。
考虑使用并查集进行缩点,然后对于所有连通块跑 0-1 背包。
AC Code
C++
#include <iostream>
#pragma GCC optimize(2)
using namespace std;
const int N = 10010;
int n, m, k;
int f[N], p[N];
int v[N], w[N];
int find(int x)
{
if (x != p[x]) p[x] = find(p[x]);
return p[x];
}
int main()
{
scanf("%d%d%d", &n, &k, &m);
for (int i = 1; i <= n; i ++ )
scanf("%d%d", &v[i], &w[i]);
for (int i = 1; i <= n; i ++ )
p[i] = i; // 并查集初始化
int a, b;
while (k -- )
{
scanf("%d%d", &a, &b);
a = find(a), b = find(b);
if (a != b)
{
v[b] += v[a], w[b] += w[a];
v[a] = 0, w[a] = 0;
p[a] = b; // 对于两个有依赖的物品缩点,并累加体积和重量
}
}
for (int i = 1; i <= n; i ++ ) // 0-1背包
for (int j = m; j >= v[i]; j -- )
f[j] = max(f[j], f[j - v[i]] + w[i]);
printf("%d\n", f[m]);
return 0;
}
最后,如果觉得对您有帮助的话,点个赞再走吧!