zoukankan      html  css  js  c++  java
  • MST最小生成树及Prim普鲁姆算法

    MST在前面学习了Kruskal算法,还有一种算法叫做Prim的。这两者的区别是Prim算法适合稠密图,比如说鸟巢这种几乎所有点都有相连的图。其时间复杂度为O(n^2),其时间复杂度与边的数目无关;而kruskal算法的时间复杂度为O(eloge),跟边的数目有关,适合稀疏图。

     

    prim算法

      基本思想:假设G=(V,E)是连通的,TE是G上最小生成树中边的集合。算法从U={u0}(u0∈V),TE={ 空集 }开始。重复执行下列操作:

       1.在所有u∈U,v∈V-U的边(u,v)∈E中找一条权值最小的边(u0,v0)并入集合TE中,同时v0并入U,直到V=U为止;接下里以 v0为边的起点,继续寻找权值最小的边并入集合TE中,依次往复;

       2.最后,TE中必有n-1条边,T=(V,TE)为G的最小生成树。

       Prim算法的核心:始终保持TE中的边集构成一棵生成树,也就是它与Kruskal算法的主要区别是,Prim是一直保持一种串联的状态而不遵从整体的贪心算法。其实初始点uo的选择可以随意,一般做题题目条件会给出或者取最小的权值边。

    其实现的代码如下:

     

    #include <stdio.h>
    #include <string.h>
    #include <stdlib.h>
    #define infinity 1000000
    #define max_vertexes 6 
    
    
    typedef int Graph[max_vertexes][max_vertexes];
    void prim(Graph G,int vcount,int father[])
    {    
    int i,j,k; 
    int lowcost[max_vertexes];
    int closeset[max_vertexes],used[max_vertexes];
    int min;  
    for (i=0;i<vcount;i++)     
      {
    /* 最短距离初始化为其他节点到1号节点的距离 */   
        lowcost[i]=G[0][i];
        /* 标记所有节点的依附点皆为默认的1号节点 */
         closeset[i]=0;      
      used[i]=0;    
        father[i]=-1;      
    }    
    used[0]=1; /*第一个节点是在s集合里的*/
    /* vcount个节点至少需要vcount-1条边构成最小生成树 */  
      for (i=1;i<=vcount-1;i++)      
       {       
     j=0;
         min = infinity;
           /* 找满足条件的最小权值边的节点k */      
         for (k=1;k<vcount;k++)
             /* 边权值较小且不在生成树中 */     
     if ((!used[k])&&(lowcost[k]<min)) 
        {
                  min =  lowcost[k];
                  j=k;
                }       
        father[j]=closeset[j];   
    printf("%d %d
    ",j+1,closeset[j]+1);//打印边   
    used[j]=1;;//把第j个顶点并入了U中     
    for (k=1;k<vcount;k++)
             /* 发现更小的权值 */       
       if (!used[k]&&(G[j][k]<lowcost[k]))       
    { 
                      lowcost[k]=G[j][k];/*更新最小权值*/       
          closeset[k]=j;;/*记录新的依附点*/
        }      
       }
    }
                     
    int main()
    {
    FILE *fr;
    int i,j,weight;
    Graph G;
    int fatheer[max_vertexes];
    for(i=0; i<max_vertexes; i++)
    for(j=0; j<max_vertexes; j++)
    G[i][j] = infinity;
    fr = fopen("prim.txt","r");
    if(!fr)
    {
    printf("fopen failed
    ");
    exit(1); 
    }
    while(fscanf(fr,"%d%d%d", &i, &j, &weight) != EOF)
    {
    G[i-1][j-1] = weight;
    G[j-1][i-1] = weight;
    }
    prim(G,max_vertexes,fatheer);
    return 0;
    }
    
    

    邻接矩阵的形式进行存储的实现:

     

    #include <stdio.h>
    #define n 6
    #define MaxNum 10000  /*定义一个最大整数*/
    
    /*定义邻接矩阵类型*/
    typedef int adjmatrix[n+1][n+1];   /*0号单元没用*/
    
    typedef struct{
    	int fromvex,tovex;
    	int weight;
    }Edge;
    typedef Edge *EdgeNode;
    
    int arcnum;     /*边的个数*/
    
    /*建立图的邻接矩阵*/
    void CreatMatrix(adjmatrix GA){
    	int i,j,k,e;
    	printf("图中有%d个顶点
    ",n);
    	for(i=1;i<=n;i++){
    		for(j=1;j<=n;j++){
    			if(i==j){
    				GA[i][j]=0;         /*对角线的值置为0*/
    			}
    			else{
    				GA[i][j]=MaxNum;    /*其它位置的值置初始化为一个最大整数*/
    			}
    		}
    	}
    	printf("请输入边的个数:");
    	scanf("%d",&arcnum);
    	printf("请输入边的信息,按照起点,终点,权值的形式输入:
    ");
    	for(k=1;k<=arcnum;k++){
    		scanf("%d,%d,%d",&i,&j,&e);  /*读入边的信息*/
    		GA[i][j]=e;
    		GA[j][i]=e;
    	}
    }
    
    /*初始化图的边集数组*/
    void InitEdge(EdgeNode GE,int m){
    	int i;
    	for(i=1;i<=m;i++){
    		GE[i].weight=0;
    	}
    }
    
    /*根据图的邻接矩阵生成图的边集数组*/
    void GetEdgeSet(adjmatrix GA,EdgeNode GE){
    	int i,j,k=1;
    	for(i=1;i<=n;i++){
    		for(j=i+1;j<=n;j++){
    			if(GA[i][j]!=0&&GA[i][j]!=MaxNum){
    				GE[k].fromvex=i;
    				GE[k].tovex=j;
    				GE[k].weight=GA[i][j];
    				k++;
    			}
    		}
    	}
    }
    
    /*按升序排列图的边集数组*/
    void SortEdge(EdgeNode GE,int m){
    	int i,j,k;
    	Edge temp;
    	for(i=1;i<m;i++){
    		k=i;
    		for(j=i+1;j<=m;j++){
    			if(GE[k].weight>GE[j].weight){
    				k=j;
    			}
    		}
    		if(k!=i){
    			temp=GE[i];GE[i]=GE[k];GE[k]=temp;
    		}
    	}
    }
    
    /*利用普里姆算法从初始点v出发求邻接矩阵表示的图的最小生成树*/
    void Prim(adjmatrix GA,EdgeNode T){
    	int i,j,k,min,u,m,w;
    	Edge temp;
    	/*给T赋初值,对应为v1依次到其余各顶点的边*/
    	k=1;
    	for(i=1;i<=n;i++){
    		if(i!=1){
    			T[k].fromvex=1;
    			T[k].tovex=i;
    			T[k].weight=GA[1][i];
    			k++;
    		}
    	}
    	/*进行n-1次循环,每次求出最小生成树中的第k条边*/
    	for(k=1;k<n;k++){
    		min=MaxNum;
    		m=k;
    		for(j=k;j<n;j++){
    			if(T[j].weight<min){
    				min=T[j].weight;m=j;
    			}
    		}
    		/*把最短边对调到k-1下标位置*/
    		temp=T[k];
    		T[k]=T[m];
    		T[m]=temp;
    		/*把新加入最小生成树T中的顶点序号赋给j*/
    		j=T[k].tovex;
    		/*修改有关边,使T中到T外的每一个顶点保持一条到目前为止最短的边*/
    		for(i=k+1;i<n;i++){
    			u=T[i].tovex;
    			w=GA[j][u];
    			if(w<T[i].weight){
    				T[i].weight=w;T[i].fromvex=j;
    			}
    		}
    	}
    }
    
    /*输出边集数组的每条边*/
    void OutEdge(EdgeNode GE,int e){
    	int i;
    	printf("按照起点,终点,权值的形式输出的最小生成树为:
    ");
    	for(i=1;i<=e;i++){
    		printf("%d,%d,%d
    ",GE[i].fromvex,GE[i].tovex,GE[i].weight);
    	}
    }
    
    void main(){
    	adjmatrix GA;
    	Edge GE[n*(n-1)/2],T[n];
    	CreatMatrix(GA);
    	InitEdge(GE,arcnum);
    	GetEdgeSet(GA,GE);
    	SortEdge(GE,arcnum);
    	Prim(GA,T);
    	printf("
    ");
    	OutEdge(T,n-1);
    }

  • 相关阅读:
    数据库高级查询
    简明python教程笔记一
    2017.12.29问题
    Windows下Python + Flask环境搭建
    ApiTestEngine框架
    接口测试框架ApiTestEngine相关
    IDEA查看JMeter源码
    IDEA安装,破解,和配置
    Uiautomator 2.0实例
    框架搭建实例
  • 原文地址:https://www.cnblogs.com/suncoolcat/p/3296980.html
Copyright © 2011-2022 走看看