POJ 1258 Agri-Net【MST】

标签：

Description

Input

Output

Sample Input

4

0 4 9 21

4 0 8 17

9 8 0 16

21 17 16 0

Sample Output

28

题目大意：john想把n个农场连接到一起，询问最小花费。给出邻接矩阵的值，求MST

思路：贪心克鲁斯卡尔算法AC

代码：

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int map[150][150];
int f[150];
struct path
{
    int x,y,w;
}a[100*100*2];
int cmp(path a,path b)
{
    return a.w<b.w;
}
int find(int a)
{
    int r=a;
    while(f[r]!=r)
    r=f[r];
    int i=a;
    int j;
    while(i!=r)
    {
        j=f[i];
        f[i]=r;
        i=j;
    }
    return r;
}
void merge(int a,int b)
{
    int A,B;
    A=find(a);
    B=find(b);
    if(A!=B)
    f[B]=A;
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        int cont=0;
        for(int i=0;i<n;i++)
        {
            f[i]=i;
        }
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                int k;
                scanf("%d",&k);
                a[cont].x=i;
                a[cont].y=j;
                a[cont++].w=k;
            }
        }
        int output=0;
        sort(a,a+cont,cmp);
        for(int i=0;i<cont;i++)
        {
            if(find(a[i].x)!=find(a[i].y))
            {
                merge(a[i].x,a[i].y);
                output+=a[i].w;
            }
        }
        printf("%d\n",output);
    }
}

POJ 1258 Agri-Net【MST】

标签：

原文地址：http://blog.csdn.net/mengxiang000000/article/details/51331207