【ybtoj 高效进阶 3.1】 【并查集】 逐个击破

【ybtoj 高效进阶 3.1】 【并查集】 逐个击破

题目

---

解题思路

换一个思路
求出保留的边
用总边权减去保留的
能被保留下来的边的两个端点不能都被k军团占领
我们要做的就是维护每个连通块
有被占领的集合代表就是被占领的，否则无所谓
按边权降序排序
判断每一条边的两个端点
如果有没被占领的就减去此条边权
两个端点如果不在同一集合
合并两个集合

---

代码

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
int n,m,k,x,p[100010],fa[100010];
long long ans;
struct lzf{
	int x,y,w;
}f[500010];

bool cmp(lzf l,lzf r)
{
	return l.w>r.w;
}

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

int main(){
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=n;i++) fa[i]=i;
	for(int i=1;i<=k;i++)
	{
		scanf("%d",&x);
		p[x+1]=1;
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&f[i].x,&f[i].y,&f[i].w);
		f[i].x++,f[i].y++;  //从1开始，不++只有60分OTATO
		ans+=f[i].w;
	}	
	sort(f+1,f+m+1,cmp);
	for(int i=1;i<=m;i++)
	{
		int xx=find(f[i].x),yy=find(f[i].y);  //查找所在集合元素
		if(xx==yy)  //同一集合不合并
			ans-=f[i].w;
		else if(!p[xx]||!p[yy])  //不在同一集合且没有都被占领，合并
		{
			ans-=f[i].w;
			if(!p[xx])fa[xx]=yy;
			else fa[yy]=xx;
		}
	}
	printf("%lld",ans);
}