hdu 3371 Connect the Cities （prime）

题目链接

/*

简单prime。。

一次AC

*/

#include"stdio.h"
#include"string.h"
#define INF 99999999
int map[502][502],dis[502],mark[502];
int n,m,k,cnt,sum;
void prime()
{
	int i,j,k,min;
	memset(mark,0,sizeof(mark));
	for(i=2;i<=n;i++)
		dis[i]=map[1][i];
	mark[1]=1;dis[1]=0;

	for(i=2;i<=n;i++)//进行n-1次！！！！
	{
		min=INF;
		for(j=2;j<=n;j++)
		{
			if(!mark[j]&&dis[j]<min)
			{
				min=dis[j];k=j;
			}
		}
		mark[k]=1;
		for(j=2;j<=n;j++)
		{
			if(!mark[j]&&dis[j]>map[k][j])
				dis[j]=map[k][j];
		}
	}
	cnt=1;sum=0;
	for(i=2;i<=n;i++)
	{
		if(dis[i]!=INF)
		{
			sum+=dis[i];
			cnt++;
		}
	}
}
int main()
{
	int i,j,t,a,c,b,aa[102];
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d%d",&n,&m,&k);
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
				map[i][j]=map[j][i]=INF;
		}
		while(m--)
		{
			scanf("%d%d%d",&a,&b,&c);
			if(map[a][b]>c)
				map[a][b]=map[b][a]=c;
		}
		while(k--)
		{
			scanf("%d",&a);
			for(i=1;i<=a;i++)
				scanf("%d",&aa[i]);
			for(i=1;i<=a;i++)
			{
				for(j=1;j<=a;j++)
					map[aa[i]][aa[j]]=map[aa[j]][aa[i]]=0;
			}
		}

		prime();
		if(cnt!=n)
			printf("-1\n");
		else
			printf("%d\n",sum);
	}
	return 0;
}