思路:刚开始想把已经修好的路,放到同一集合中去,不过发现这样做在查找根节点时,会出错,我就把已修好的路的花费设为0,放到优先队列中,不过这样挺浪费时间滴。
C++提交卡过,G++, TL
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<queue>
using namespace std;
struct node
{
int st,ed,w;
bool friend operator <(const node &a,const node &b)
{
return a.w>b.w;
}
};
int far[511];
int rank[511];
int n,m,k;
int s;
priority_queue<node>Q;
void init()
{
for(int i=0; i<=n; i++)
{
far[i]=i;
rank[i]=1;
}
}
int find_set(int x)
{
if(far[x]==x)
return x;
far[x]=find_set(far[x]);
return far[x];
}
int kruskal(int nn)
{
int sum=0;
node tem;
while(!Q.empty())
{
tem=Q.top();
Q.pop();
int x=find_set(tem.st);
int y=find_set(tem.ed);
if(x!=y)
{
if(rank[x]>rank[y])
{
far[y]=x;
rank[x]+=rank[y];
}
else
{
far[x]=y;
rank[y]+=rank[x];
}
sum+=tem.w;
s++;
}
}
return sum;
}
int main()
{
int T;
int p,q,c;
node tem;
scanf("%d\n",&T);
while(T--)
{
while(!Q.empty())
Q.pop();
s=0;
scanf("%d%d%d",&n,&m,&k);
init();
for(int i=0; i<m; i++)
{
scanf("%d%d%d",&p,&q,&c);
tem.st=p;
tem.ed=q;
tem.w=c;
Q.push(tem);
}
int gh;
for(int i=0; i<k; i++)
{
scanf("%d",&gh);
scanf("%d",&p);
for(int j=1; j<gh; j++)
{
scanf("%d",&q);
tem.st=p;
tem.ed=q;
tem.w=0;
Q.push(tem);
}
}
int ans=kruskal(n);
if(s!=n-1)
printf("-1\n");
else
printf("%d\n",ans);
}
return 0;
}