CSU 1326: The contest 并查集+分组背包
Description
殷犇有很多队员。他们都认为自己是最强的,于是,一场比赛开始了~
于是安叔主办了一场比赛,比赛有n个题目,每个题目都有一个价值Pi和相对能力消耗Wi,但是有些题目因为太坑不能同时做出来,并且坑题具有传递性。(a和b一起做会坑、b和c会坑则a和c也会坑) ACM队员们想知道,于是他们想知道在能力范围内,它们最多可以作出多少价值的题目。
聪明的你,告诉我,能帮帮他们吗?
Input
第1行两个整数,n,Wmax,k(0<=n,Wmax,k<=1000),其中n为题目总数,Wmax为初始的总能力数.
接下来n行,为每个题目的Pi,Wi(0<=Pi<=1000,1<=Wi<=10,均为整数)
再接下来k行,每行2个数字a,b表示a和b会坑
Output
对每组数据输出1行为最大可能价值
Sample Input
3 10 1 100 1 200 5 10 5 1 2
Sample Output
210
Hint
每个测试点1s
CSU_WJQ
Source
CSU Monthly 2013 Oct.思路: 利用并查集将题目分成一组 分组背包
#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
int dp[1005];
struct node{
int cost,value;
};
node a[1005];
int d[1005];
int n,wmax,k;
int father(int x)
{
if(d[x] == -1)
return x;
else return d[x] = father(d[x]);
}
void union_(int x,int y)
{
int root1 = father(x);
int root2 = father(y);
if(root1 != root2)
d[root1] = root2;
}
vector<node> vec[1005];
int vis[1005];
int main()
{
while(scanf("%d%d%d",&n,&wmax,&k) != EOF)
{
for(int i = 1 ; i <= n ; i++)
scanf("%d%d",&a[i].value,&a[i].cost);
memset(d,-1,sizeof(d));
memset(vis,0,sizeof(vis));
memset(dp,0,sizeof(dp));
for(int i = 1 ; i <= k ; i++)
{
int x,y;
scanf("%d%d",&x,&y);
union_(x,y);
vis[x] = vis[y] = 1;
}
for(int i = 0 ; i <= n ; i++)
vec[i].clear();
for(int i = 1 ; i <= n ; i++)
{
if(vis[i] == 0)
{
vec[i].push_back(a[i]);
}else{
vec[father(i)].push_back(a[i]);
}
}
for(int i = 1 ; i <= n ; i++)
if(vec[i].size() != 0)
for(int j = wmax ; j >= 0 ; j--)
for(int k = 0 ; k < vec[i].size() ; k++)
{
if(vec[i][k].cost > j)
continue;
else
dp[j] = max(dp[j],dp[j-vec[i][k].cost] + vec[i][k].value);
}
printf("%d\n",dp[wmax]);
}
return 0;
}