POJ 1258（最小生成树+知识）

用kruskal算法，利用w[i]给r[i]间接排序，从而r[i]可以按照边大小保存序号，同时要判断是否在一个集合里面

#include <cstdio>
#include <iostream>
#include <queue>
using namespace std;

#define sf scanf
#define pf printf
#define debug printf("!\n")
#define blank printf("\n")
#define mem(a,b) memset(a,b,sizeof(a))

const int MaxN = ;
const int INF = <<;

int p[MaxN];

int map[MaxN][MaxN],w[MaxN*MaxN],r[MaxN*MaxN],u[MaxN*MaxN],v[MaxN*MaxN];

int m,n;

int find(int x){return p[x]==x?x:p[x]=find(p[x]);}

int cmp(const int a,const int b)
{
    return w[a]<w[b];
}

int kruskal()
{
    int ans = ,i;

    for(i = ;i<n;i++) p[i] = i;
    for(i = ;i<m;i++) r[i] = i;

    sort(r,r+m,cmp);

    for(i = ;i<m;i++)
    {
        int e = r[i];
        int x = find(u[e]);
        int y = find(v[e]);
        if(x!=y)
        {
            ans+=w[e];p[x] = y;
        }
    }
    return ans;
}

int main()
{
    int i,j;
    while(~sf("%d",&n))
    {
        m = ;
        mem(u,);
        mem(v,);
        mem(w,);
        for(i = ;i<n;i++)
        {
            for(j=;j<n;j++)
                sf("%d",&map[i][j]);
        }
        for(i = ;i<n;i++)
        {
            for(j=i+;j<n;j++)
            {
                u[m] = i;
                v[m] = j;
                w[m++] = map[i][j];
            }
        }
        int ans = kruskal();
        pf("%d\n",ans);

    }

    return ;
}