hdu-4313-并查集

http://acm.hdu.edu.cn/showproblem.php?pid=4313

题意，给你一个n点，n-1条边的图

给你k个点，

要求删除一些边使得k个点完全不连通

求删除边集的最小价值

又是这种删边的问题，可以逆向地用并查集处理

正向地一条条删边是很麻烦的，如果我们逆过来，从头构图，把边按权值由大到小排序，每次选一条边，看加入 其后【是否导致 K个点间有联通】，若无则可选之，如此重复到最后即可得到 最小价值的 【使得k个点不联通的边集】

合并的时候，若有一方父亲 属于K个点中的点，则优先选其为父亲，便于判断 【是否导致 K个点间有联通】

#include <cstdio>
#include <cmath>
#include <cstring>
#include <string>
#include <algorithm>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <iostream>
using namespace std;

const double pi=acos(-1.0);
double eps=0.000001;

int fa[100005];
int find(int x)
{
    if (fa[x]==x)
        return x;
    else return fa[x]=find(fa[x]);
}
struct node
{
    int x,y,z;
    node() {}
    node(int a,int b,int c)
    {
        x=a,y=b,z=c;
    }
};
bool cmp(node a,node b)
{
    return a.z>b.z;
}
node tm[100005];
int main()
{
    int t;
    cin>>t;

    while(t--)
    {
        set<int >sb;
        sb.clear();
        int n,k;
        cin>>n>>k;
                long long ans=0;
        for (int i=0;i<=n;i++) fa[i]=i;
        for (int i=1; i<n; i++)
        {
            scanf("%d %d %d",&tm[i].x,&tm[i].y,&tm[i].z);
            ans+=tm[i].z;
        }
        sort(tm+1,tm+n,cmp);
        for (int i=1; i<=k; i++)
        {
            int tmp;
            scanf("%d",&tmp);
            sb.insert(tmp);
        }
        long long sum=0;
        for (int i=1; i<n; i++)
        {
            int fx=find(tm[i].x);
            int fy=find(tm[i].y);
            if (sb.find(fx)!=sb.end()&&sb.find(fy)!=sb.end()) continue;  //必不可选的情况
            if (sb.find(fx)!=sb.end()||sb.find(fy)!=sb.end())<span style="white-space:pre">		</span>//选择把父亲为， K个点中的点
            {
                if (sb.find(fx)!=sb.end())
                    fa[fy]=fx;
                        else fa[fx]=fy;
            }
            else<span style="white-space:pre">		</span>//任意选父亲
            {
                fa[fx]=fy;
            }
            sum+=tm[i].z;
        }
        printf("%lld\n",ans-sum);
    }


    return 0;
}