【happy birthday】CSU 1326 The contest【并查集+分组背包】

1326: The contest

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 680   Solved: 213
[ Submit][ Status]

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.

思路：

1、很明显的分组背包。我们用并查集将一起做会坑的题放在一个组别中。

2、然后利用01背包的特性，我们将分组背包按照一组一组处理即可。

Ac代码：

#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;
struct node
{
    int cost,val;
}a[1005];
int dp2[1050];
int dp[1050];
int f[1005];
vector<int >mp[1005];
int find(int a)
{
    int r=a;
    while(f[r]!=r)
    r=f[r];
    int i=a;
    int j;
    while(i!=r)
    {
        j=f[i];
        f[i]=r;
        i=j;
    }
    return r;
}
int merge(int a,int b)
{
    int A,B;
    A=find(a);
    B=find(b);
    if(A!=B)
    {
        f[B]=A;
    }
}
int main()
{
    int n,v,m;
    while(~scanf("%d%d%d",&n,&v,&m))
    {
        for(int i=1;i<=n;i++)mp[i].clear();
        for(int i=1;i<=n;i++)f[i]=i;
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&a[i].val,&a[i].cost);
        }
        for(int i=0;i<m;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            merge(x,y);
        }
        for(int i=1;i<=n;i++)
        {
            mp[find(i)].push_back(i);
        }
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=1005;i++)
        {
            memset(dp2,0,sizeof(dp2));
            for(int j=0;j<=1005;j++)dp2[j]=dp[j];
            for(int j=0;j<mp[i].size();j++)
            {
                int num=mp[i][j];
                for(int k=v;k>=a[num].cost;k--)
                {
                    dp2[k]=max(dp2[k],dp[k-a[num].cost]+a[num].val);
                }
            }
            for(int j=0;j<=1005;j++)dp[j]=max(dp[j],dp2[j]);
        }
        printf("%d\n",dp[v]);
    }
}