HDU 4313 Matrix

题意：  n个点的一棵树，其中有k个敌人所在结点，要破坏一些边使得这 k 个点互不可达，求出破坏边的最小权值和。

分析：  对于有n个结点的树，删除任意的k (k<=n-1)条边都能将原树分成k+1个部分

          按照题意至少需要将原树划分成 k 个部分（此时每部分中都包含一个敌人的点），删除的边数为k-1。

           类似kruskal最小生成树的过程，不过此处将边按权值从大到小排列，每次将边加进来时要判断是否会使两个危险的

           点连通，是的话这条边就是需要被删除的，否则将它加到树上。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
#define maxn 100005
#define clr(x)memset(x,0,sizeof(x))
int n,m;
int u[maxn],v[maxn],w[maxn];
int r[maxn],f[maxn];
int find(int x)
{
    return f[x]==x?x:(f[x]=find(f[x]));
}
void join(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    {
        if(fx<fy)
            f[fy]=fx;
        else f[fx]=fy;
    }
}
bool cmp(int a,int b)
{
    return w[a]>w[b];
}
int pos[maxn];
__int64 kruskal()
{
    int i;
    __int64 res=0;
    for(i=0;i<n;i++)
        f[i]=i;
    for(i=0;i<n-1;i++)
        r[i]=i;
    sort(r,r+n-1,cmp);
    for(i=0;i<n-1;i++)
    {
        int e=r[i];
        int fx=find(u[e]);
        int fy=find(v[e]);
        if(pos[fx]&&pos[fy])
        {
            res+=w[e];
            continue;
        }
        if(pos[fx]||pos[fy])
            pos[fx]=pos[fy]=1;
        join(fx,fy);
    }
    return res;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        int i,a,b,c;
        for(i=0;i<n-1;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            u[i]=a;
            v[i]=b;
            w[i]=c;
        }
        clr(pos);
        for(i=0;i<m;i++)
        {
            int tmp;
            scanf("%d",&tmp);
            pos[tmp]=1;
        }
        printf("%I64d\n",kruskal());
    }
    return 0;
}