poj 1258 AgriNet 最小生成树

这题跟上一个题是一样的。就出所有最小生成树的所有边的值就可以了。

直接代码吧。

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
#define MAXN 10000
#define MAXM 50000
using namespace std;
int point[500+10][500+10];
struct edge
{
    int u,v,w;
} edges[MAXM];
int parent[MAXN];
int n,m;
int i,j;
int maxx;

//并查集
void UFset()
{

    for(i=1; i<=n; i++)
        parent[i]=-1;
}
int Find(int x)
{
    int s;
    for(s=x; parent[s]>=0; s=parent[s])  ;
    while(s!=x)
    {
        int tmp=parent[x];
        parent[x]=s;
        x=tmp;
    }
    return s;
}
void Union(int R1,int R2)
{
    int r1=Find(R1),r2=Find(R2);
    int tmp=parent[r1]+parent[r2];
    if(parent[r1]>parent[r2])
    {
        parent[r1]=r2;
        parent[r2]=tmp;
    }
    else
    {
        parent[r2]=r1;
        parent[r1]=tmp;
    }
    return;
}
//并查集
bool cmp(edge a,edge b)
{
    return a.w<b.w;
}
void Kruskal()
{
    maxx=0;
    int sumweight=0;
    int num=0;
    int u,v;
    UFset();
    for(i=0; i<m; ++i)
    {
        u=edges[i].u;
        v=edges[i].v;
        if(Find(u)!=Find(v))
        {
            if(edges[i].w>maxx) maxx=edges[i].w;
            sumweight+=edges[i].w;
            ++num;
            Union(u,v);
        }
    }
    printf("%d\n",sumweight);
}

int main()
{
    int t,count;
    while(scanf("%d",&n)!=EOF)
    {
        memset(point,0,sizeof(point));



        for(i=0; i<n; ++i)
            for(j=0; j<n; ++j)
            {
                scanf("%d",&point[i][j]);
                if(point[i][j]==0) point[i][j]=1000000000;
            }
        m=0;
        for(i=0; i<n; ++i)
            for(j=0; j<=i; ++j)
            {
                if(i==j) continue;
                edges[m].u=i+1;
                edges[m].v=j+1;
                edges[m].w=point[i][j];
                m++;
            }
        sort(edges,edges+m,cmp);
        Kruskal();

    }
    return 0;
}