一步一步写算法(之prim算法 下)

【 声明:版权所有,欢迎转载,请勿用于商业用途。  联系信箱:feixiaoxing @163.com】


    前两篇博客我们讨论了prim最小生成树的算法,熟悉了基本的流程。基本上来说,我们是按照自上而下的顺序来编写代码的。首先我们搭建一个架构,然后一步一步完成其中的每一个子功能,这样最后构成一个完成prim算法计算过程。


    f)将DIR_LINE队列中不符合的数据删除,主要是双节点都已经访问过的DIR_LINE数据。

void delete_unvalid_line_from_list(DIR_LINE** ppHead, MINI_GENERATE_TREE* pMiniTree)
{
	DIR_LINE* prev;
	DIR_LINE* pcur;
	STATUS result;

	prev = NULL;
	pcur = *ppHead;
	while(pcur){
		if(!check_valid_for_line(pcur, pMiniTree)){
			result = delete_line_from_queue(ppHead, pcur);
			assert(TRUE == result);

			if(NULL == prev)
				pcur = *ppHead;
			else
				pcur = prev->next;

			continue;
		}

		prev = pcur;
		pcur = pcur->next;
	}

	return;
}

     g) 在f)函数中使用了判定DIR_LINE合法性的函数,我们需要完善一下。

int check_valid_for_line(DIR_LINE* pDirLine, MINI_GENERATE_TREE* pMiniTree)
{
	int index;
	int flag_start;
	int flag_end;

	flag_start = 0;
	flag_end = 0;

	for(index = 0; index < pMiniTree->node_num; index ++){
		if(pDirLine->start == pMiniTree->pNode[index]){
			flag_start = 1;
			break;
		}
	}

	for(index = 0; index < pMiniTree->node_num; index ++){
		if(pDirLine->end == pMiniTree->pNode[index]){
			flag_end = 1;
			break;
		}
	}

	return (1 == flag_start && 1 == flag_end) ? 0 : 1;
}

     h) 最后就是对当前已经入队的DIR_LINE数据排序,其实就是 链表排序  

void insert_for_sort_operation(DIR_LINE** ppNode, DIR_LINE* pNode)  
{  
    DIR_LINE* prev;  
    DIR_LINE* cur;  
	
    /* 在第一个数据之前插入pNode */  
    if(pNode->weight < (*ppNode)->weight){  
        pNode->next = *ppNode;  
        *ppNode = pNode;  
        return;  
    }  
	
    cur = *ppNode;  
    while(cur){  
        if(pNode->weight < cur->weight)  
            break;  
		
        prev = cur;  
        cur = cur->next;  
    }  
	
    pNode->next = prev->next;  
    prev->next = pNode;  
    return;  
} 

void sort_for_line_list(DIR_LINE** ppNode)
{
	DIR_LINE* prev;
	DIR_LINE* curr;
	
	if(NULL == ppNode || NULL == *ppNode)
		return;
	
	curr = (*ppNode) ->next;
	(*ppNode) ->next = NULL;
	
	while(curr){
		prev = curr;
		curr = curr->next;
		insert_for_sort_operation(ppNode, prev);
	}
}	

算法总结:

    1)算法本身还有改进的空间,比如是不是内存分配上每一次都要重建DIR_LINE队列有待商榷

    2)算法编写不是一部就位的,中间有反复更有删改,写四五次是很正常的事情

    3)编写代码的时候最好做到边修改、边测试,这样可以一方面增加代码的健壮度,一方面还能提高自己的信心

    4)如果存在可能,可以复用以前写过的、稳定的算法代码,比如说排序、查找、堆栈、二叉树之类的代码


  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
Prim算法是一种求无向加权连通图的最小生成树的贪心算法,可以用C语言来实现。下面是一份示例代码: ```c #include <stdio.h> #include <stdlib.h> #include <limits.h> #define MAX_VERTICES 1000 // 最大顶点数 typedef struct { int v1, v2, weight; } Edge; typedef struct { int count; int parent[MAX_VERTICES]; int key[MAX_VERTICES]; int mst[MAX_VERTICES][MAX_VERTICES]; } Graph; Graph create_graph(int vertices) { Graph G; G.count = vertices; for (int i = 0; i < vertices; i++) { for (int j = 0; j < vertices; j++) { G.mst[i][j] = 0; } G.key[i] = INT_MAX; G.parent[i] = -1; } return G; } void add_edge(Graph *G, int v1, int v2, int weight) { G->mst[v1][v2] = weight; G->mst[v2][v1] = weight; } int extract_min(Graph *G, int *visited) { int min = INT_MAX, min_index = -1; for (int i = 0; i < G->count; i++) { if (visited[i] == 0 && G->key[i] < min) { min = G->key[i]; min_index = i; } } return min_index; } void prim(Graph *G) { int visited[MAX_VERTICES] = {0}; G->key[0] = 0; for (int i = 0; i < G->count-1; i++) { int u = extract_min(G, visited); visited[u] = 1; for (int v = 0; v < G->count; v++) { if (G->mst[u][v] != 0 && visited[v] == 0 && G->mst[u][v] < G->key[v]) { G->parent[v] = u; G->key[v] = G->mst[u][v]; } } } for (int i = 1; i < G->count; i++) { add_edge(G, G->parent[i], i, G->mst[G->parent[i]][i]); } } void print_mst(Graph G) { for (int i = 1; i < G.count; i++) { printf("(%d, %d) -> %d\n", G.parent[i], i, G.mst[G.parent[i]][i]); } } int main() { Graph G = create_graph(6); add_edge(&G, 0, 1, 4); add_edge(&G, 0, 2, 3); add_edge(&G, 1, 2, 1); add_edge(&G, 1, 3, 2); add_edge(&G, 2, 3, 4); add_edge(&G, 3, 4, 2); add_edge(&G, 4, 5, 6); prim(&G); print_mst(G); return 0; } ``` 这份代码实现了Prim算法,使用邻接矩阵来表示图,其中`create_graph`函数用来创建一个空的图,`add_edge`函数用来添加一条边,`extract_min`函数用来从尚未访问的顶点中选取key值最小的顶点,`prim`函数用来求出最小生成树,`print_mst`函数用来输出最小生成树的边。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

嵌入式-老费

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值