最小生成树之Prim算法---POJ1258---Agri-Net

题目链接:http://poj.org/problem?id=1258

最小生成树的模板题/(ㄒoㄒ)/~~

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int N=110;
int map[N][N];
int vis[N];
int low[N];
int T;
int pos;
int prim()
{
    int min,res=0,j;
    for(int i=1;i<T;i++)
    {
        min=1000000;
        for( j=1;j<=T;j++)
        {
            if(vis[j]==0&&min>low[j])
            {
                min=low[j];
                pos=j;
            }
        }
        res+=min;
        vis[pos]=1;
        for(int q=1;q<=T;q++) { if(vis[q]==0&&low[q]>map[pos][q])
            {
                low[q]=map[pos][q];
            }
        }
    }
    return res;
}
int main()
{
    while(scanf("%d",&T)!=EOF)
    {
        memset(vis,0,sizeof(vis));
        memset(low,0,sizeof(low));
        for(int i=1;i<=T;i++)
        {
            for(int j=1;j<=T;j++)
            {
                scanf("%d",&map[i][j]);
            }
        }
        for(int i=1;i<=T;i++)
        {
            low[i]=map[1][i];
        }
        pos=1;
        vis[1]=1;
        int ans=prim();
        printf("%d\n",ans);
    }
    return 0;
}

仅代表个人观点,欢迎交流探讨,勿喷~
最小生成树之Prim算法---POJ1258---Agri-Net_第1张图片
PhotoBy:WLOP

http://weibo.com/wlop

你可能感兴趣的:(算法,Prim)