hdu 3371 Connect the Cities 最小生成树prim

#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define N 550
#define INF 0x7ffffff
using namespace std;

int d
,v
,mp

,n;

int prim()
{
for(int i=1;i<=n;i++)   d[i]=INF,v[i]=0;
d[1]=0;
for(int i=0;i<n;i++)
{
int mmin=INF,t=-1;
for(int j=1;j<=n;j++)
if(!v[j]&&d[j]<mmin)
mmin=d[t=j];
v[t]=1;
if(t==-1)   return -1;
for(int j=1;j<=n;j++)
if(!v[j])
d[j]=min(d[j],mp[t][j]);
}
int ans=0;
for(int i=1;i<=n;i++)   ans+=d[i];
return ans;
}

int main()
{
int T,m,k;
cin>>T;
while(T--)
{
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
mp[i][j]=INF;
for(int i=0;i<m;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
if(mp[u][v]>w)  mp[u][v]=mp[v][u]=w;
}
for(int i=0;i<k;i++)
{
int t,u,v;
scanf("%d",&t);
for(int j=0;j<t;j++)
{
scanf("%d",&u);
if(j)   mp[u][v]=mp[v][u]=0;
v=u;
}
}
cout<<prim()<<endl;
}
}