逐个击破（最小生成树）

逐个击破

题目背景

三大战役的平津战场上，傅作义集团在以北平、天津为中心，东起唐山西至张家口的铁路线上摆起子一字长蛇阵，并企图在溃败时从海上南逃或向西逃窜。为了就地歼敌不让其逃走，指挥官制定了先切断敌人东西两头退路然后再逐个歼灭敌人的战略方针。秉承伟大军事家的战略思想，作为一个有智慧的军长你，遇到了一个类似的战场局面。

题目描述

现在有 $N$ 个城市，其中 $K$ 个被敌方军团占领了，$N$ 个城市间有 $N-1$ 条公路相连，破坏其中某条公路的代价是已知的，现在，告诉你 $K$ 个敌方军团所在的城市，以及所有公路破坏的代价，请你算出花费最少的代价将这 $K$ 个地方军团互相隔离开，以便第二步逐个击破敌人。

输入格式

第一行包含两个正整数 $N$ 和 $K$。

第二行包含 $K$ 个整数，表示哪个城市被敌军占领。

接下来 $N-1$ 行，每行包含三个正整数 $a,b,c$，表示从 $a$ 城市到 $b$ 城市有一条公路，以及破坏的代价 $c$。城市的编号从 $0$ 开始。

输出格式

输出一行一个整数，表示最少花费的代价。

样例 #1

样例输入 #1

5 3
1 2 4
1 0 4
1 3 8
2 1 1
2 4 3

样例输出 #1

4

提示

对于 $10\%$ 的数据，$N\le 10$。

对于 $100\%$ 的数据，$2\le N\le 10^5$，$2\le K\le N$，$1\le c\le 10^6$。

思路

我们知道，破坏公路是比较复杂的（因为我们感觉都是做建边的题，很少做过删边的题），那么我们正难则反，把删边变成加边。而且：还要满足删的边最便宜，就只要选贵的加就是了。那么此时就可以用kruskal了·。

代码

#include<iostream>
#include<algorithm>
#include<cstring>

#define int long long

using namespace std;

const int N = 1e5+10;

struct E{
    int x,y,w;
}e[N];
bool cmp(E a,E b){
    return a.w>b.w;
}
int p[N];
int n,m;
int w[N];
bool st[N];

int find(int x){
    if(x!=p[x])p[x]=find(p[x]);
    return p[x];
}

signed main(){
    cin>>n>>m;
    
    for(int i=1;i<=n;i++)p[i]=i;
    for(int i=1;i<=m;i++){
        cin>>w[i];
        st[w[i]]=true;
    }
    int sum=0;//正难则反，把删边变成了加边
    //还要满足删的边最便宜，就只要选贵的加就是了
    for(int i=1;i<n;i++){
        int a,b,c;
        cin>>a>>b>>c;
        e[i]={a,b,c};
        sum+=c;
    }
    
    sort(e+1,e+1+n,cmp);
    
    for(int i=1;i<n;i++){
        int pa=find(e[i].x),pb=find(e[i].y),w=e[i].w;
        if(st[pa]&&st[pb])continue;
        p[pa]=pb;
        sum-=w;
        st[pa]=st[pb]=st[pa]|st[pb];
    }
    
    cout<<sum;
    
    return 0;
}