【ybt高效进阶3-1-6】【luogu P2700】逐个击破

逐个击破

题目链接：ybt高效进阶3-1-6 / luogu P2700

题目大意

有 n 个点，m 条边，每条边有价值。要你删去总价值尽可能小的边，使得一些规定的点不能相互连通。

思路

这道题我们考虑反着来考虑，变成保留总价值尽可能多的边使得规定点不连通。

那我们可以想到用并查集维护这个点所在的连通块中有没有定的点。
那我们贪心，从最大的边开始连，能连就连。
那怎样能连呢？要么已经连通，那在连也没关系。要么两边都没有规定的点或这有一边有。

那你如何维护这个连通块里面有没有规定的点呢？
那我们可以让那个点一直是这个连通块的最父亲，那如果一个连通块有规定的点，那它里面的点的最父亲一定是被标记了是规定的点的。

那就这样并查集，读入边时加上所有边的权值。最后输出所有边的权值减去你算出的最大值即可。

代码

#include<cstdio>
#include<algorithm>

using namespace std;

struct node {
	int x, y;
	long long z;
}a[500010];
int n, m, k, x;
int fa[100010], cant[100010];
long long ans;

bool cmp(node x, node y) {
	return y.z < x.z; 
}

int find(int now) {
	if (fa[now] == now) return now;
		else return fa[now] = find(fa[now]);
}

void connect(int x, int y) {
	if (cant[x]) fa[y] = x;
		else fa[x] = y;
	//如果一边有规定的点就一定是它那边做父亲，否则任意一边都可以
}

int main() {
	scanf("%d %d %d", &n, &m, &k);
	//这里给的是ybtoj的代码，如果要luogu的就不都如m，把m赋值为n-1即可
	
	for (int i = 1; i <= n; i++) fa[i] = i;
	
	for (int i = 1; i <= k; i++) {
		scanf("%d", &x);
		cant[x + 1] = 1;
	}
	
	for (int i = 1; i <= m; i++) {
		scanf("%d %d %lld", &a[i].x, &a[i].y, &a[i].z);
		a[i].x++;
		a[i].y++;
		ans += a[i].z;
	}
	
	sort(a + 1, a + m + 1, cmp);//按权值从大到小排序
	
	for (register int i = 1; i <= m; i++) {
		int X = find(a[i].x), Y = find(a[i].y);
		if (X == Y) {//已经连通，就不管，可以直接连
			ans -= a[i].z;
		}
		else if (!cant[X] || !cant[Y]) {//不会两边都有规定的点
			ans -= a[i].z;
			connect(X, Y);
		}
	}
	
	printf("%lld", ans);
	
	return 0;
}