题目来源:vjudge.net/contest/241341#problem/D
题意就不用说了,挺好理解的。
思路:用prim算法求最小生成树就行了。
ac代码:
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<stack>
#include<cstdio>
#include<map>
#include<set>
#include<string>
#include<queue>
using namespace std;
#define inf 0x3f3f3f3f
const int maxn=205;
int mp[maxn][maxn];
int visit[maxn];
int main()
{
int i,j,t,n,m,k;
int s,ans,ls=0;
int a,b,c,mins,minc,mins1,minc1;
int i1;
scanf("%d",&t);
while(t--)
{
memset(visit,0,sizeof(visit));
ls++;
ans=0;
for(i=0;i<maxn;i++)
for(j=0;j<maxn;j++)
mp[i][j]=inf;
scanf("%d%d%d",&n,&m,&k);
for(i=1;i<=k;i++)
{
scanf("%d",&s);
visit[s]=1;
}
for(i=0;i<m;i++)
{
scanf("%d%d%d",&a,&b,&c);
mp[a][b]=c;
mp[b][a]=c;
}
while((++k)<=n)
{
mins1=inf;
for(i=1;i<=n;i++)
{
if(!visit[i])
{
mins=inf;
for(j=1;j<=n;j++)
{
if(visit[j]&&mins>mp[i][j])
{
mins=mp[i][j];
}
}
if(mins<mins1)
{
mins1=mins;
minc1=i;
}
}
}
// printf("%d\n",minc1);
ans+=mins1;
visit[minc1]=1;
}
printf("Case #%d: %d\n",ls,ans);
}
return 0;
}