HDU 4313 Matrix(并查集)

时间:2024-01-16 12:56:56

http://acm.hdu.edu.cn/showproblem.php?pid=4313

题意:

给出一棵树,每条边都有权值,其中有几个点是特殊点,现在破坏边还使得这几个特殊点互相不可达,需要使得破坏的边的权值和最小。

思路:
解法很妙!

利用并查集,先将每个点分成一个集合,将边按照降序排序,如果该边两边所在的集合没有特殊点或者只有一边有,那么就可以不用删,用并查集合并。如果两边都有特殊点的话,那就只能删处这条边,因为已经排好序了,所以删的边是尽量最小的。

 #include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = ;
typedef long long ll; int n, m;
int vis[maxn], p[maxn]; struct node
{
int u,v,w;
bool operator< (const node& rhs) const
{
return w > rhs.w;
}
}a[maxn]; int finds(int x)
{
return x==p[x]?x:p[x]=finds(p[x]);
} int main()
{
//freopen("in.txt","r",stdin);
int T;
scanf("%d",&T);
while(T--)
{
memset(vis,,sizeof(vis));
scanf("%d%d",&n,&m);
for(int i=;i<n;i++) p[i] = i;
for(int i=;i<n-;i++) scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].w);
for(int i=;i<m;i++)
{
int x; scanf("%d",&x);
vis[x] = ;
}
sort(a,a+n-);
ll ans = ;
for(int i=;i<n-;i++)
{
int u = a[i].u;
int v = a[i].v;
int x = finds(u);
int y = finds(v);
if(x!=y && (!vis[x]||!vis[y]))
{
p[x] = y;
if(vis[x]||vis[y]) vis[y] = vis[x] = ;
}
else ans += a[i].w;
}
printf("%lld\n",ans);
}
return ;
}