Networking---poj1287最小生成树

http://poj.org/problem?id=1287

最小生成树模板题类似的还有：poj1258  hdu1233代码几乎一样；

最小生成树详解

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<map>
#define N 110
#define INF 0xfffffff
using namespace std;

int ans, n, m, dist[N], maps[N][N], vis[N];//dist[i]代表当前起点到i的最小距离；

void Init()
{
    for(int i=0; i<=n; i++)
    {
        dist[i] = INF;
        vis[i] = 0;
        for(int j=0; j<=n; j++)
        if(i==j)
            maps[i][j] = 0;
        else
            maps[i][j] = INF;
    }
}

void Prim(int start)
{
    for(int i=1; i<=n; i++)
        dist[i] = maps[start][i];
    vis[start] = 1;
    for(int i=1; i<=n; i++)
    {
        int Min = INF, index=-1;

        for(int j=1; j<=n; j++)
        {
            if(vis[j]==0 && Min>dist[j])
            {
                Min = dist[j];
                index = j;
            }
        }
        if(index == -1)break;
        vis[index] = 1;
        ans += Min;
        for(int j=1; j<=n; j++)
        {
            if(vis[j]==0 && maps[index][j] < dist[j])
            {
                dist[j] = maps[index][j];
            }
        }
    }
}

int main()
{
    int a, b, c;
    while(scanf("%d", &n),n)
    {
        ans=0;
        Init();
        scanf("%d", &m);
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d", &a, &b, &c);
            maps[a][b] = maps[b][a] = min(maps[a][b], c);
        }
        Prim(1);
        printf("%d
", ans);
    }
    return 0;
}

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<map>
#define N 110
#define INF 0xfffffff
using namespace std;

int n, f[N], m;

struct node
{
    int x, y, d;
}a[N*N];

int cmp(node p, node q)
{
    return p.d < q.d;
}

int Find(int x)
{
    if(x != f[x])
        f[x] = Find(f[x]);
    return f[x];
}

int main()
{
    int ans, px, py;
    while(scanf("%d", &n),n)
    {
        ans=0;
        for(int i=0; i<=n; i++)
            f[i] = i;
        scanf("%d", &m);
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d", &a[i].x, &a[i].y, &a[i].d);
        }
        sort(a, a+m, cmp);
        for(int i=0; i<m; i++)
        {
            px = Find(a[i].x);
            py = Find(a[i].y);
            if(px != py)
            {
                f[px] = py;
                ans+=a[i].d;
            }
        }
        printf("%d
", ans);
    }
    return 0;
}