TOJ--3073:Country Road 最小生成树（Prim）

//TOJ--3073:Country Road     最小生成树（Prim）
#include<iostream>
#include<memory.h>
#define MAX 1003
#define INF 1010
using namespace std;

int map[MAX][MAX],visited[MAX];
int dist[MAX];
int N,M,K;

int Prim()
{
int i,j,pos,min,sum;

for(i=1;i<=N;i++)
dist[i]=map[1][i];//dist[]初始化为从起点到各点的距离。

visited[1]=1;

for(i=1;i<N;i++)//总共N个节点，已经把第一个节点放进去了，剩下还得放N-1个节点
{
min=INF;
for(j=1;j<=N;j++)
{
if(!visited[j] && dist[j]<min)//找与刚加入的点距离最小的点
{
min=dist[j];
pos=j;
}
}

visited[pos]=1;//将pos点加入

for(j=1;j<=N;j++)//更新dist[]
{
if(!visited[j] && map[pos][j]<dist[j])
{
dist[j]=map[pos][j];
}
}
}

sum=0;
for(i=1;i<=N;i++)
{
sum=sum+dist[i];
if(dist[i]==INF)
return -1;
}
return sum;
}

int main()
{
int caseNum;
int i,j,s,t,d;
cin>>caseNum;
while(caseNum--)
{
cin>>N>>M>>K;
memset(visited,0,sizeof(visited));

for(i=1;i<=N;i++)//初始化map
{
for(j=1;j<=N;j++)
{
if(i==j)   map[i][j]=0;//节点到自身的距离设为0
else       map[i][j]=INF;//节点到其他节点的距离设为一个比较大的值（改值要比所有节点间最大权值大）
}
}

for(i=1;i<=M;i++)
{
cin>>s>>t;
map[s][t]=0;//输入已经有路径的节点对，并将有路径可达的节点间距离设为0
map[t][s]=0;//注意：无向图，双向赋值
}

for(i=1;i<=K;i++)
{
cin>>s>>t>>d;
map[s][t]=d;//若在节点s和t间修路，则距离为d
map[t][s]=d;
}
cout<<Prim()<<endl;
}
return 0;
}