prim算法,轉過來當模版用

轉載自:http://www.cnblogs.com/Veegin/archive/2011/04/29/2032388.html


#include <stdio.h>
#include <string.h>
#define MaxInt 0x3f3f3f3f
#define N 110
//創建map二維數組儲存圖表,low數組記錄每2個點間最小權值,visited數組標記某點是否已訪問
int map[N][N],low[N],visited[N];
int n;
 
int prim()
{
    int i,j,pos,min,result=0;
    memset(visited,0,sizeof(visited));
//從某點開始,分別標記和記錄該點
    visited[1]=1;pos=1;
//第一次給low數組賦值
    for(i=1;i<=n;i++)
        if(i!=pos) low[i]=map[pos][i];
//再運行n-1次
    for(i=1;i<n;i++)
    {
//找出最小權值並記錄位置
     min=MaxInt;
     for(j=1;j<=n;j++)
         if(visited[j]==0&&min>low[j])
         {
             min=low[j];pos=j;
         }
//最小權值累加
    result+=min;
//標記該點
    visited[pos]=1;
//更新權值
    for(j=1;j<=n;j++)
        if(visited[j]==0&&low[j]>map[pos][j])
            low[j]=map[pos][j];
    }
    return result;
}
 
int main()
{
    int i,v,j,ans;
    while(scanf("%d",&n)!=EOF)
    {
//所有權值初始化爲最大
        memset(map,MaxInt,sizeof(map));
        for(i=1;i<=n;i++)
            for(j=1;j<=n;j++)
            {
                scanf("%d",&v);
                map[i][j]=map[i][j]=v;
            }
            ans=prim();
            printf("%d\n",ans);
    }
    return 0;
}


發佈了31 篇原創文章 · 獲贊 40 · 訪問量 8萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章