hdu 4313(类似于kruskal）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4313

思路：初始时一条边都不加，将所有边按权值从大到小排序，判断每一个边两端的顶点是否是均为machine节点，如果是则应删除这条边（即sum要加上这条边的权值），否则加入这条边，然后在并查集合并时尽量让根节点为machine节点。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define MAXN 100010
typedef long long ll;
int n,k;
int parent[MAXN];
bool mark[MAXN];
struct Edge{
   int u,v,w;
}edge[MAXN];
ll sum;

int cmp(const Edge &p,const Edge &q){
   return p.w>q.w;
}

int Find(int x){
   int s;
   for(s=x;s!=parent[s];s=parent[s])
   ;
   while(s!=x){
      int tmp=parent[x];
      parent[x]=s;
      x=tmp;
   }
   return s;
}

void Union(int u,int v){
   if(mark[u])parent[v]=u;
   else if(mark[v])parent[u]=v;
   else if(u<v)parent[v]=u;
   else parent[u]=v;
}

int main(){
   int _case,x;
//   freopen("1.txt","r",stdin);
   scanf("%d",&_case);
   while(_case--){
      scanf("%d%d",&n,&k);
      for(int i=0;i<n-1;i++){
         scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
         parent[i]=i;
      }
      parent[n-1]=n-1;
      sort(edge,edge+n-1,cmp);
      memset(mark,false,sizeof(mark));
      for(int i=1;i<=k;i++){ scanf("%d",&x);mark[x]=true; }
      sum=0;
      for(int i=0;i<n-1;i++){
         int u=Find(edge[i].u);
         int v=Find(edge[i].v);
         int w=edge[i].w;
         if(mark[u]&&mark[v]){
            sum+=w;
         }else
            Union(u,v);
      }
      printf("%I64d\n",sum);
   }
   return 0;
}