洛谷：P2700 逐个击破（最大生成树、贪心）

先断退路，逐个歼灭

题意：

• 很清晰，不再赘述

思路：

• 一眼看过去，显然可以发现很多性质：

若一条边两点（连通块）都存在军团，显然这条边会被算入结果
若一条边两点都不存在军团，显然可以通过缩点维护一个 min { 连通块内最小边权 }

如果只是根据这些性质用并查集缩点，最后树会变成一颗 “黑白树”，即任意相邻两点一个是军团一个不是。（也就是二分图）

但在此之后处理还是蛮麻烦，需要 dfs 去找两个军团间的区间。

• 所以我们大胆贪心猜想、尝试，先把所有边按权值从大到小排序，再如上分析情况：

若一条边两点（连通块）都存在军团，显然这条边会被算入结果
若一条边两点有一个不存在军团，这条边可以先不计入答案。因为边从大到小排序，你尽管缩点，直到两连通块都存在军团时，此时的边权肯定最小，对答案的贡献也最优

$Code:$

#include<bits/stdc++.h>
#include<unordered_set>
#include<unordered_map>
#define mem(a,b) memset(a,b,sizeof a)
#define cinios (ios::sync_with_stdio(false),cin.tie(0),cout.tie(0))
#define sca scanf
#define pri printf
#define ul (u << 1)
#define ur (u << 1 | 1)
#define fx first
#define fy second
//#pragma GCC optimize(2)
//[博客地址](https://blog.csdn.net/weixin_51797626?t=1) 
using namespace std;

typedef long long ll;
typedef pair<ll, int> PII;
typedef pair <ll, PII> PI;

const int N = 100010, M = 4000010, MM = N;
int INF = 0x3f3f3f3f, mod = 100003;
ll LNF = 0x3f3f3f3f3f3f3f3f;
int n, m, k, T, S, D;
int p[N];
bool st[N];
struct edge
{
	int a, b, w;
	bool operator <(const edge& ee)const { return w > ee.w; }
}ed[N];

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

int main() {
	cinios;

	cin >> n >> k;
	for (int i = 0; i < n; i++)p[i] = i;

	for (int i = 0; i < k; i++) {
		cin >> m;
		st[m] = true;
	}

	m = n - 1;
	for (int i = 0; i < m; i++) {
		int a, b, x;
		cin >> a >> b >> x;
		ed[i] = { a,b,x };
	}
	sort(ed, ed + m);

	ll ans = 0;
	for (int i = 0; i < m; i++) {
		int fa = find(ed[i].a), fb = find(ed[i].b), w = ed[i].w;

		if (st[fa] && st[fb])ans += w;
		else {
			st[fb] = st[fa] | st[fb];//维护军团信息
		}
		p[fa] = fb;
	}

	cout << ans;

	return 0;
}
/*
*/