HDU 4313 并查集

点击打开链接

题意：给一个城镇的图，有m个城镇上是有敌兵的，为了将所有敌兵的联系隔断，需要删除的所有边的最小的权值

思路：看了就知道是并查集Kruskal思想的题，我们将边的权值从大到小向里面加，如果我要加的这条变加进去之后，敌兵可以相连，那么这条边肯定要删下去，而我们从大到小加的边，所以肯定是最小的，因为可以的边我们都用上了嘛

#include <queue>
#include <vector>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int V,E;
int par[100005],ran[100005],vis[100005];
void init(int n){
int i;
for(i=0;i<=n;i++){
ran[i]=0;
par[i]=i;
}
}
int find(int x){
if(par[x]==x)
return x;
return par[x]=find(par[x]);
}
void unite(int x,int y){
x=find(x);
y=find(y);
par[x]=y;
}
bool same(int x,int y){
return find(x)==find(y);
}
struct node{
int u,v,cost;
};
bool cmp(node a,node b){
return a.cost>b.cost;
}
node es[100005];
long long kruskal(){
int i;
long long res=0;
init(V);
sort(es,es+E,cmp);
for(i=0;i<E;i++){
node e=es[i];
int x=find(e.u),y=find(e.v);
if(vis[x]&&vis[y]) res=res+(long long)e.cost;
else if(vis[x]) unite(e.v,e.u);
else if(vis[y]) unite(e.u,e.v);
else unite(e.u,e.v);
}
return res;
}
int main(){
int t,m,i,j,ans,sum,a;
scanf("%d",&t);
while(t--){
scanf("%d%d",&V,&m);
memset(vis,0,sizeof(vis));
E=V-1;
for(i=0;i<E;i++){
scanf("%d%d%d",&es[i].u,&es[i].v,&es[i].cost);
}
for(i=0;i<m;i++){
scanf("%d",&a);
vis[a]=1;
}
printf("%I64d\n",kruskal());
}
return 0;
}