LA 6437 Power Plant (prim最小生成树)

还是裸的最小生成树

#include<bits/stdc++.h>
using namespace std;
int T,N,M,P,K,a,b,c;
int dist[1020],m[1020][1020];
bool p[1020];
void prim()
{
for(int i=1;i<=N;i++)
{
if(!p[i])
{
int Min=1000020;
for(int j=1;j<=N;j++)
{
if(p[j]&&m[i][j]!=0&&m[i][j]<Min)
Min=m[i][j];
}
dist[i]=Min;
}
}
for(int i=1;i<=N-K;i++)
{
int Min=INT_MAX,k=0;
for(int j=1;j<=N;j++)
{
if(!p[j]&&dist[j]<Min)
{
Min=dist[j];
k=j;
}
}
if(k==0)
return;
p[k]=true;
for(int j=1;j<=N;j++)
{
if(!p[j]&&m[k][j]!=0&&dist[j]>m[k][j])
dist[j]=m[k][j];
}
}
}
int main()
{
scanf("%d",&T);
for(int kase=1;kase<=T;kase++)
{
memset(p,false,sizeof(p));
memset(m,0,sizeof(m));
scanf("%d%d%d",&N,&M,&K);
for(int i=1;i<=K;i++)
{
scanf("%d",&P);
dist[P]=0;
p[P]=true;
}
for(int i=1;i<=M;i++)
{
scanf("%d%d%d",&a,&b,&c);
m[a][b]=c;
m[b][a]=c;
}
prim();
int ans=0;
for(int i=1;i<=N;i++)
ans+=dist[i];
printf("Case #%d: %d\n",kase,ans);
}
return 0;
}