http://acm.hdu.edu.cn/showproblem.php?pid=3371用C++能过，G++过不了不过用prim能过

关注

原创最新推荐文章于 2022-03-16 18:32:58 发布 894 阅读 AI 写同款· GEO 优化 ›

AI权益加码！Claude Code、Cursor等20+工具免费用！购周边限时加赠Coding Plan Lite，畅享主流AI工具！学习进阶更高效！阅读详情

	#include<cstdio>
	#include<algorithm>
	#include<string.h>
	#include<functional>
	#include<cmath>
	#include<iostream>
	using namespace std;
	class node
	{
	public:
	int u,v;
	int dist;
	}edge[55002];
	int num[509];
	int find(int x)
	{
	return x==num[x]?x:num[x]=find(num[x]);
	}
	bool cmp(node xx,node yy)
	{
	return xx.dist<yy.dist;
	}
int main()
{
	int t;
	scanf("%d",&t);
 while(t--)
   {
	  int n,m,k;
	 scanf("%d%d%d",&n,&m,&k);
    for(int i=0;i<=508;i++)
	    {
	     num[i]=i;
	    }
	int a,b,x,mn=1,nn,nx,mx;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&a,&b,&x);
		edge[i].u=a;edge[i].v=b;edge[i].dist=x;
	}
	 while(k--)
	 {
		 int tt;
		 int con[510]={0};
	   scanf("%d",&tt);
	   for(int i=0;i<tt;++i)
	   {
	    scanf("%d",&con[i]);
		  
	   }
	   for(int i=1;i<tt;++i)
		   num[find(con[i])]=find(con[i-1]);//连接幸存的道路
	 }
	 int xn=1;
	 for(int i=1;i<=n;i++)
		 for(int j=1;j<=n;j++)//xn为幸存的道路数
		 {
		    if(i==j)
				continue;
			if(num[j]==i)
				xn++;
		 }
	sort(edge+1,edge+1+m,cmp);
	int sum=0;
	
	for(int i=1;xn<n&&i<=m;i++)
	{
		int cx=find(edge[i].u),cy=find(edge[i].v);
		if(cx!=cy)
		{
			num[cx]=cy;
			sum+=edge[i].dist;
			xn++;
		}
	}
	if(xn<n)//因为初始化为1所以如果能连通，xn==n
		sum=-1;
	printf("%d\n",sum);
	}
	return 0;
}

收录于