hdu 4313 Matrix （最小生成树krusual）

题目链接

题目意思：有多个城市，多条路，路都是双向的，有一些城市有机器人大军，我们想通过破坏城市之间的城市来断绝机器人大军的联系，并且用最少的时间。

用krusual算法，并用mark标记存在有机器人的城市，有的两个城市之间的道路只有的1个机器人大军，剩下的就是要破坏的边。。。

#include"stdio.h"#include"stdlib.h"#include"string.h"int mark[100001],set[100001];__int64 ans;struct node{int x,y,t;}aa[100001];int find(int x){if(x==set[x])return x;return set[x]=find(set[x]);}int cmp(const void*a,const void*b){return (*(struct node*)b).t-(*(struct node*)a).t;}void merge(int x,int y){int fx,fy;fx=find(x);fy=find(y);if(fx==fy)return ;if(fx<fy){set[fy]=fx;mark[fx]+=mark[fy];}else {set[fx]=fy;mark[fy]+=mark[fx];}}int main(){int T,n,m,i,a,b,c,k,x,y;scanf("%d",&T);while(T--){scanf("%d%d",&n,&m);memset(mark,0,sizeof(mark));for(i=0;i<=n;i++)set[i]=i;ans=0;k=0;for(i=0;i<n-1;i++){scanf("%d%d%d",&a,&b,&c);aa[k].x=a;aa[k].y=b;aa[k].t=c;k++;ans+=c;}qsort(aa,k,sizeof(aa[0]),cmp);for(i=0;i<m;i++){scanf("%d",&a);mark[a]=1;}for(i=0;i<k;i++){x=find(aa[i].x);y=find(aa[i].y);//两个城市之间最多有一个城市有机器人if(mark[x]+mark[y]<=1){merge(x,y);ans-=aa[i].t;}}printf("%I64d\n",ans);}return 0;}