poj 1258 最小生成树 模板

POJ  最小生成树模板
Kruskal算法
 
#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
#include<ctype.h>
#include<stdlib.h>
#include<limits.h>
#include<math.h>
#include<queue>
#include<stack>
#define max(a, b) a>b?a:b;
#define min(a, b) a<b?a:b;
using namespace std;
const int N = ;
 
typedef struct rode
{
    int x, y, l;
} RODE;
RODE maps[N];
int f[N];
 
int cmp(const void *a, const void *b)
{
    RODE *c, *d;
    c=(RODE *)a;
    d=(RODE *)b;
    return c -> l - d -> l;
}
 
void Kruskal(int k);
int Find(int x);
 
int main()
{
    int n, i, j, k, a;
    while(scanf("%d", &n)!=EOF)
    {
        k=;
        for(i=; i<n; i++)
        {
            f[i]=i;
        }
        for(i=; i<n; i++)
            for(j=; j<n; j++)
            {
                scanf("%d", &a);
                maps[k].x=i;
                maps[k].y=j;
                maps[k].l=a;
                k++;
            }
        qsort(maps, k, sizeof(maps[]), cmp);
        Kruskal(k);
    }
    return ;
}
void Kruskal(int k)
{
    int ans=;
    for(int i=; i<k; i++)
    {
        int ru=Find(maps[i].x);
        int rv=Find(maps[i].y);
        if(ru!=rv)
        {
            f[ru]=rv;
            ans+=maps[i].l;
        }
    }
    printf("%d\n", ans);
}
int Find(int x)
{
    if(f[x]!=x)
        f[x]=Find(f[x]);
    return f[x];
}
 
另一种  prim算法 和dijkstra算法很像
 
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
 
using namespace std;
 
int Map[][];
const int oo=0xfffffff;
int visit[];
int dist[];
int prim(int s,int n)
{
    int ans=;
    memset(visit,,sizeof(visit));
    for(int i=; i<=n; i++)
        dist[i]=Map[s][i];
    visit[s]=;
    for(int i=; i<n; i++)
    {
        int index;
        int Min=oo;
        for(int j=; j<=n; j++)
        {
            if(!visit[j]&&dist[j]<Min)
            {
                index=j;
                Min=dist[j];
            }
        }
        ans+=Min;
        visit[index]=;
        for(int j=; j<=n; j++)
        {
            if(!visit[j]&&dist[j]>Map[index][j])
            {
                dist[j]=Map[index][j];
            }
        }
 
    }
    return ans;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        for(int i=; i<=n; i++)
        {
            for(int j=; j<=n; j++)
            {
                Map[i][j]=oo;
            }
        }
 
        for(int i=; i<=n; i++)
        {
            for(int j=; j<=n; j++)
            {
                scanf("%d",&Map[i][j]);
            }
        }
        printf("%d\n",prim(,n));
    }
    return ;
}