快捷導(dǎo)航

最小生成樹(shù)算法之Prim算法

更新時(shí)間：2015年07月23日 10:53:01 作者：砍柴1990

這篇文章主要講解了普里姆算法（Prim算法），圖論中的一種算法，可在加權(quán)連通圖里搜索最小生成樹(shù),需要的朋友可以參考下

本文介紹了最小生成樹(shù)的定義，Prim算法的實(shí)現(xiàn)步驟，通過(guò)簡(jiǎn)單舉例實(shí)現(xiàn)了C語(yǔ)言編程。

1.什么是最小生成樹(shù)算法？
簡(jiǎn)言之，就是給定一個(gè)具有n個(gè)頂點(diǎn)的加權(quán)的無(wú)相連通圖，用n-1條邊連接這n個(gè)頂點(diǎn)，并且使得連接之后的所有邊的權(quán)值之和最小。這就叫最小生成樹(shù)算法，最典型的兩種算法就是Kruskal算法和本文要講的Prim算法。

2.Prim算法的步驟是什么？
這就要涉及一些圖論的知識(shí)了。
a.假定圖的頂點(diǎn)集合為V，邊集合為E.
b.初始化點(diǎn)集合U={u}.//u為V中的任意選定的一點(diǎn)
c.從u的鄰接結(jié)點(diǎn)中選取一點(diǎn)v使這兩點(diǎn)之間的權(quán)重最小,然后將v加入集合U中.
d.從結(jié)點(diǎn)v出發(fā)，重復(fù)c步驟，直到V={}.

3.舉個(gè)例子來(lái)說(shuō)明Prim算法的步驟：
一個(gè)簡(jiǎn)單的加權(quán)拓?fù)鋱D如下所示

選取1為初始點(diǎn)，則按照上面所示的步驟訪問(wèn)結(jié)點(diǎn)的順序依次次為：

則最終訪問(wèn)結(jié)點(diǎn)的順序：1,3,4,2,5.
4.Prim算法的具體C語(yǔ)言編程實(shí)現(xiàn)：

#include <stdio.h>
#include <cstdlib>
#include<memory.h>
const int Max =0x7fffffff;
const int N=50;
 
int n;
int g[N][N],dis[N],visited[N];
 
int prim()
{
  int i,j;
  int pos,min;
  int ans=0;
  memset(visited,0,sizeof(visited));
  visited[1]=1;pos=1;
  //assign a value to the dis[N] first
  for(i=2;i<=n;i++)
    dis[i]=g[pos][i];
  for(i=1;i<n;i++)
  {
    min=Max; 
    for(j=1;j<=n;j++)
    {
      if(visited[j]==0&&min>dis[j])
      {
        min=dis[j];
        pos=j; 
      }
    }
    printf("The node being traversed is :%d\n",pos);
    ans+=min;
    printf("The value of ans is %d\n",ans);
    //mark the node
    visited[pos]=1;
    //update the weight
    for(j=1;j<=n;j++)
      if(visited[j]==0&&dis[j]>g[pos][j])
        dis[j]=g[pos][j];
  }
  return ans;
}
 
int main()
{
  int i=1,j=1;
  int ans=0;
  int w;
  printf("Please enter the number of the nodes:\n");
  scanf("%d",&n);
  for(i=1;i<=n;i++)
    for(j=1;j<=n;j++)
    {
      if(i==j)
        g[i][j]=0;
      else
        g[i][j]=Max;
    }
  printf("Please enter the number of the edges:\n");
  int edgenum;
  scanf("%d",&edgenum);
  int v1,v2;
  printf("Please enter the number and the corresponding weight:\n");
  for(i=1;i<=edgenum;i++)
  {
    scanf("%d%d%d",&v1,&v2,&w);
    g[v1][v2]=g[v2][v1]=w;
  }
  ans=prim();
  printf("The sum of the weight of the edges is:%d\n",ans);
  system("pause");
  return 0;
   
}

5.程序運(yùn)行后的結(jié)果截圖