hdu 4313 Matrix 并查集

题目分析：给n个顶点，和n-1条边，生成一个树，给定k个点，，要求去掉几条边之后，要求这k个顶点不在一个集合内,并且所删除的边得权值之和最小，分析后很明显删除k-1条边，

思路：先对边进行降序排序，然后插入边，如果该边的两个顶点都是属于这k个顶点，那么这条边要删去

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int set[100100];
int vis[100100];

struct node
{
int s,e,v;
}arr[100100];

int cmp(node x,node y)
{
return  x.v>y.v;
}
int find(int x)
{
int i;
for(i=x;i!=set[i];i=set[i]);
while(i!=x)
{
int temp=set[x];
set[x]=i;
x=temp;
}
return i;
}
int main()
{
int T,n,k;
scanf("%d",&T);
while(T--)
{
int ans=0;
memset(vis,0,sizeof(vis));
scanf("%d %d",&n,&k);
for(int i=0;i<n-1;i++)
scanf("%d %d %d",&arr[i].s,&arr[i].e,&arr[i].v);

for(int i=0;i<n;i++)
set[i]=i;

for(int i=0;i<k;i++)
{
int temp;
scanf("%d",&temp);
vis[temp]=1;
}
sort(arr,arr+n,cmp);
for(int i=0;i<n-1;i++)
{

int fx=find(arr[i].s);
int fy=find(arr[i].e);

if(vis[fx]==1&&vis[fy]==1)//if(vis[fx]==1&&vis[fx]==1)
{
ans+=arr[i].v;
}
else if(vis[fx]==1)
{
set[fy]=fx;
}
else if(vis[fy]==1)
{
set[fx]=fy;
}
else if(fx<fy)
{
set[fy]=fx;
}
else
{
set[fx]=fy;
}
}
printf("%d\n",ans);
}
system("pause");
return 0;
}

/*
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

int set[100100];
struct node
{
int s,e,v;
}arr[100100];
int cmp(node x,node y)
{
return  x.v>y.v;
}
int find(int x)
{
int i;
for(i=x;set[i]!=-1&&i!=set[i];i=set[i]);
while(i!=x)
{
if(set[x]==-1)
break;
int temp=set[x];
set[x]=i;
x=temp;
}
return i;
}
int main()
{
int T,n,k;
scanf("%d",&T);
while(T--)
{
int ans=0;
scanf("%d %d",&n,&k);
for(int i=0;i<n-1;i++)
scanf("%d %d %d",&arr[i].s,&arr[i].e,&arr[i].v);

for(int i=0;i<n;i++)
set[i]=i;

for(int i=0;i<k;i++)
{
int temp;
scanf("%d",&temp);
set[temp]=-1;
}
for(int i=0;i<n;i++)
printf("set[%d]=%d*\n",i,set[i]);
sort(arr,arr+n,cmp);

for(int i=0;i<n-1;i++)
{
printf("g*");
int fx=find(arr[i].s);
if(arr[i].s==2)
printf("set[2]=%d***\n\n",fx);
printf("g*");
int fy=find(arr[i].e);
if(arr[i].e==1)
printf("set[1]=%d***\n\n",fy);
printf("g*");

if(fx==-1&&fy==-1)
{
ans+=arr[i].v;
}
else if(fx<fy)
{
set[fy]=fx;
}
else
{
set[fx]=fy;
}
}
for(int i=0;i<n;i++)
printf("set[%d]=%d\n",i,set[i]);
printf("%d\n",ans);
}
system("pause");
return 0;
}
*/