hdu 4313(类似于kruskal）
题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4313
思路：初始时一条边都不加，将所有边按权值从大到小排序，判断每一个边两端的顶点是否是均为machine节点，如果是则应删除这条边（即sum要加上这条边的权值），否则加入这条边，然后在并查集合并时尽量让根节点为machine节点。
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 using namespace std; 6 #define MAXN 100010 7 typedef long long ll; 8 int n,k; 9 int parent[MAXN]; 10 bool mark[MAXN]; 11 struct Edge{ 12 int u,v,w; 13 }edge[MAXN]; 14 ll sum; 15 16 int cmp(const Edge &p,const Edge &q){ 17 return p.w>q.w; 18 } 19 20 int Find(int x){ 21 int s; 22 for(s=x;s!=parent[s];s=parent[s]) 23 ; 24 while(s!=x){ 25 int tmp=parent[x]; 26 parent[x]=s; 27 x=tmp; 28 } 29 return s; 30 } 31 32 void Union(int u,int v){ 33 if(mark[u])parent[v]=u; 34 else if(mark[v])parent[u]=v; 35 else if(u<v)parent[v]=u; 36 else parent[u]=v; 37 } 38 39 int main(){ 40 int _case,x; 41 // freopen("1.txt","r",stdin); 42 scanf("%d",&_case); 43 while(_case--){ 44 scanf("%d%d",&n,&k); 45 for(int i=0;i<n-1;i++){ 46 scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w); 47 parent[i]=i; 48 } 49 parent[n-1]=n-1; 50 sort(edge,edge+n-1,cmp); 51 memset(mark,false,sizeof(mark)); 52 for(int i=1;i<=k;i++){ scanf("%d",&x);mark[x]=true; } 53 sum=0; 54 for(int i=0;i<n-1;i++){ 55 int u=Find(edge[i].u); 56 int v=Find(edge[i].v); 57 int w=edge[i].w; 58 if(mark[u]&&mark[v]){ 59 sum+=w; 60 }else 61 Union(u,v); 62 } 63 printf("%I64d\n",sum); 64 } 65 return 0; 66 }
