CodeForces - 742D Arpa's weak amphitheater and Mehrdad's valuable Hoses（并查集+分组背包）

题目链接：https://codeforces.com/problemset/problem/742/D

题意：给定n个人和m条关系，每个人都有一个重量和一个美丽值，问在不超过最大重量的情况下美丽值的最大值，其中一个朋友圈里的要么全部选，要么只选一个。

思路：这个问题可以转化为分组背包问题。把一个朋友圈看成一组，其中的“物品”便是每一个人以及所有人合在一起，这些我们都可以用并查集处理出来，然后再套一个分组背包，问题便迎刃而解啦。

具体细节请看代码：

``
#include
#include
#include
#include

using namespace std;

struct Good
{
int w,b;
Good(int _w=0,int _b=0):w(_w),b(_b) {}
};

const int maxn=1010;

vectorgood[maxn];

int f[maxn],valw[maxn],valb[maxn];
int w[maxn],b[maxn];
int F[maxn];

void init(int n)
{
for(int i=1; i<=n; i++)
{
f[i]=i;
valw[i]=w[i],valb[i]=b[i];
good[i].clear();
}
}

int Find(int x)
{
if(x==f[x]) return x;
else return f[x]=Find(f[x]);
}

void Merge(int x,int y)
{
int t1=Find(x),t2=Find(y);
if(t1!=t2)
{
f[t2]=t1;
valw[t1]+=valw[t2],valb[t1]+=valb[t2];
}
}

int main()
{
int n,m,W;
while(~scanf("%d%d%d",&n,&m,&W))
{
for(int i=1; i<=n; i++)
{
scanf("%d",&w[i]);
}
for(int i=1; i<=n; i++)
{
scanf("%d",&b[i]);
}
init(n);
while(m–)
{
int x,y;
scanf("%d%d",&x,&y);
Merge(x,y);
}

    for(int i=1; i<=n; i++)
    {
        int flag=1;
        if(Find(i)==i)
        {
            good[Find(i)].push_back(Good(valw[i],valb[i]));
            flag=0;
        }
        if((flag==0&&w[i]!=valw[i]&&b[i]!=valb[i]) || flag==1) good[Find(i)].push_back(Good(w[i],b[i]));
    }
    memset(F,0,sizeof(F));

    for(int i=1; i<=n; i++)
    {
        if(good[i].size()==0) continue;
        for(int j=W; j>=0; j--)
        {
            for(int k=0; k<good[i].size(); k++)
            {
                if(j-good[i][k].w>=0)
                {
                    F[j]=max(F[j],F[j-good[i][k].w]+good[i][k].b);
                }
            }
        }
    }
    printf("%d\n",F[W]);
}
return 0;

}