学步园

并查集
初始时没有边加入，将所有边按权值从大到小排序，然后依次判断每一个边两端的顶点是否是均为machine节点，

如果是则应删除这条边，否则加入这条边，然后在并查集合并时尽量让根节点为machine节点。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm> 
using namespace std;
#define N 100100
struct edge
{
	int s,d;
	int cost;
}e[N];
int num=0,n,f[N];
bool vis[N];
int find(int a)
{
	if(a!=f[a])
		f[a]=find(f[a]);
	return f[a];
}
bool cmp(struct edge a,struct edge b)
{
	return a.cost>b.cost;
}	
void addedge(int a,int b,int c)
{
	e[num].s=a;e[num].d=b;
	e[num++].cost=c;
}
int main()
{
	int i,j,k,t,a,b,c;
	__int64 sum;
	scanf("%d",&t);
	while(t--)
	{
		num=0;
		scanf("%d%d",&n,&k);
		memset(vis,false,sizeof(vis));
		for(i=0;i<n;i++)
			f[i]=i;
		for(i=1;i<n;i++)
		{
			scanf("%d%d%d",&a,&b,&c);
			addedge(a,b,c);
		}
		for(i=0;i<k;i++)
		{
			scanf("%d",&a);
			vis[a]=true;
		}
		sort(e,e+num,cmp);
		sum=0;
		for(i=0;i<num;i++)
		{
			a=find(e[i].s);b=find(e[i].d);
			if(a!=b&&vis[a]==true&&vis[b]==true)
			{
				sum+=e[i].cost;
			}
			else 
			{
				if(vis[a])
					f[b]=a;
				else
					f[a]=b;	
			}
		}
		printf("%I64d\n",sum);
	}
	return 0;
}