HDU 4313 Matrix（并查集）

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

题意：

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

思路：
解法很妙！

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

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 100005;
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,0,sizeof(vis));
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++)  p[i] = i;
for(int i=0;i<n-1;i++)  scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].w);
for(int i=0;i<m;i++)
{
int x; scanf("%d",&x);
vis[x] = 1;
}
sort(a,a+n-1);
ll ans = 0;
for(int i=0;i<n-1;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] = 1;
}
else ans += a[i].w;
}
printf("%lld\n",ans);
}
return 0;
}