2018年10月17日提高组 T1 平津战役

版权声明：转载无所谓的。。。 https://blog.csdn.net/xuxiayang/article/details/83187359

大意

给定 $n$个节点，现在要删除总代价最小的边使得指定的 $m$个点两两不连通

---

思路

不考虑拆边，考虑建边

因为我们要使删除总代价尽量少，也就是保留的边尽量大，也就是生成树，排序+并查集即可

---

代码

#include<cstdio>
#include<algorithm>
#define N 100001
using namespace std;long long ans;
int n,k,f[N],a,b;
bool vis[N];
inline int fa(register int x){return x==f[x]?x:f[x]=fa(f[x]);}//并查集
struct node{int from,to,w;}e[N];
inline bool cmp(node x,node y){return x.w>y.w;}
inline int read()//快读
{
	char c;int d=1,f=0;
	while(c=getchar(),c<48||c>57)if(c=='-')d=-1;f=(f<<3)+(f<<1)+c-48;
	while(c=getchar(),c>47&&c<58)f=(f<<3)+(f<<1)+c-48;
	return d*f;
}
signed main()
{
	n=read();k=read();f[n]=n;
	for(register int i=1;i<=k;i++) vis[read()]=1;
	for(register int i=1;i<n;i++) e[i]=(node){read(),read(),read()},f[i]=i,ans+=e[i].w;
	sort(e+1,e+n,cmp);
	for(register int i=1;i<n;i++)
	{
		a=fa(e[i].from);b=fa(e[i].to);
		if(vis[a]&vis[b]) continue;//若在此之前已经连接了则不管它
		f[a]=f[b];
		ans-=e[i].w;
		vis[a]=vis[b]=vis[a]|vis[b];//连接
	}
	printf("%lld",ans);
}