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;
}