CSU 1326: The contest（并查集+分组背包）

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 
#include 
#include 
#include 
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 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;
}