【洛谷P2700】逐个击破

生成树的抖机灵绿题 还是要想一下

洛谷上搜生成树的绿题刷刷 蓝题做不动
看题目：

P2700 逐个击破

题目背景

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

题目描述

现在有 \(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\le10^5\)，\(2\le K\le N\)，\(1\le c\le 10^6\)。

解法&&个人感想

拿到这道题的时候我就在想 我们要怎么删边，或者怎么建边，来实现代价最小？
都标上最小生成树了，又是这个数据范围，八成是贪心 Kruscal
使得每个店互不连通，那如果我们用数组记录“好点”和“不好点”之间的连通性呢？
对于根节点（相对并查集），去更新两个点之间不好点的连通性
如果查出来两棵树的根节点都是坏的，那么就不连
就素这样 其实也不难
哦对了 建边的时候，要从大到小，因为删掉的要最小
下面看代码:

#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define maxn 100005
#define maxm 100005
using namespace std;
struct Edge{
    int from,to,w;
};
Edge edge[maxm];
int fa[maxn];
int p[maxn];
int pd[maxn];//为1表示与x连通 反之则不连通
int n,k;
ll ans=0;
bool cmp(Edge x,Edge y){
    return x.w>y.w;
}
int get(int x){
    if(fa[x]==x) return x;
    return fa[x]=get(fa[x]);
}
void merge(int x,int y){
    fa[x]=y;
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i=0;i<=n-1;i++) fa[i]=i;
    for(int i=1;i<=k;i++){
        scanf("%d",&p[i]);
        pd[p[i]]=1;
    }
    for(int i=1;i<=n-1;i++){
        scanf("%d%d%d",&edge[i].from,&edge[i].to,&edge[i].w);
        ans+=edge[i].w;
    }
    sort(edge+1,edge+n,cmp);
    for(int i=1;i<=n-1;i++){
        int x=get(edge[i].from),y=get(edge[i].to);
        if(pd[x]&&pd[y]) continue;
        ans-=edge[i].w;
        merge(x,y);
        if(pd[x]) pd[y]=1;
        if(pd[y]) pd[x]=1;
    }
    printf("%lld\n",ans);
    system("pause");
    return 0;
}

绿题还要多刷 这个思路想了有一会