POJ 1287 Networking (ZOJ 1372) MST

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=55;
const int MAXN=3000;
int n,sum;
int fa[N];
struct dot
{
	int x,y;
	int len;
}data[MAXN];

bool operator < (const dot & a,const dot &b)
{
	return a.len<b.len;
}

int find(int cur)
{
	return cur==fa[cur]? cur: fa[cur]= find(fa[cur]);
}

int main()
{
	while(scanf("%d",&n),n)
	{
		int num;
		scanf("%d",&num);
		for(int i=0;i<num;i++)
		{
			scanf("%d%d%d",&data[i].x,&data[i].y,&data[i].len);
		}

		sort(data,data+num);

		for(int i=0;i<=n;i++)
			fa[i]=i;

		sum=0;
		for(int i=0;i<num;i++)
		{
			int root_x=find(data[i].x);
			int root_y=find(data[i].y);
			if(root_x!=root_y)
			{
				sum+=data[i].len;
				fa[root_x]=root_y;
			}
		}
		printf("%d
",sum);
	}
	return 0;
}

#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXN=55;
const int INF=999999;
int dis[MAXN],map[MAXN][MAXN];
int sum;
int n,num;
void prim()  
{  
    bool vis[MAXN]={0};  
  
    for(int i=1;i<=n;i++)  
            dis[i]=INF;  
  
    int cur=1;  
    vis[cur]=1;  
    dis[cur]=0;  
  
    for(int i=1;i<=n;i++)  
    {  
        int mini=INF;  
        for(int j=1;j<=n;j++)  
            if(!vis[j] &&  dis[j] > map[cur][j])//            
                dis[j]=map[cur][j];  
  
        for(int j=1;j<=n;j++)  
            if(!vis[j] && mini> dis[j])  
                mini=dis[cur=j];  
        vis[cur]=1;  
    }  
  
}  
int main()
{

	while(scanf("%d",&n),n)
	{
		for(int i=0;i<MAXN;i++)
			for(int j=0;j<MAXN;j++)
				map[i][j]=INF;
		scanf("%d",&num);
		for(int i=0;i<num;i++)
		{
			int x,y,len;
			scanf("%d%d%d",&x,&y,&len);

			if(map[x][y] > len)
				map[y][x]=map[x][y]=len;
		}
		sum=0;
		prim();
		 for(int i=1;i<=n;i++)  
            sum+=dis[i];  
		printf("%d
",sum);
	}
	return 0;
}