码迷,mamicode.com
首页 > 编程语言 > 详细

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

时间:2014-12-11 12:04:57      阅读:206      评论:0      收藏:0      [点我收藏+]

标签:style   blog   http   ar   color   sp   for   strong   on   

原文: 一步一步写算法(之prim算法 中)

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


    C)编写最小生成树,涉及创建、挑选和添加过程

MINI_GENERATE_TREE* get_mini_tree_from_graph(GRAPH* pGraph)
{
	MINI_GENERATE_TREE* pMiniTree;
	DIR_LINE pDirLine;

	if(NULL == pGraph || NULL == pGraph->head)
		return NULL;

	pMiniTree = (MINI_GENERATE_TREE*)malloc(sizeof(MINI_GENERATE_TREE));
	assert(NULL != pMiniTree);
	memset(pMiniTree, 0, sizeof(MINI_GENERATE_TREE));

	pMiniTree->node_num = 1;
	pMiniTree->pNode = (int*)malloc(sizeof(int) * pGraph->count);
	memset(pMiniTree->pNode, 0, sizeof(int) * pGraph->count);
	pMiniTree->pNode[0] = pGraph->head->start;

	while(1){
		memset(&pDirLine, 0, sizeof(DIR_LINE));
		get_dir_line_from_graph(pGraph, pMiniTree, &pDirLine);
		if(pDirLine.start == 0)
			break;

		pMiniTree->line_num ++;
		insert_line_into_queue(&pMiniTree->pLine, pDirLine.start, pDirLine.end, pDirLine.weight);
		insert_node_into_mini_tree(&pDirLine, pMiniTree);
	}

	return pMiniTree;
}

    d) 构建挑选函数,选择最合适的边

void get_dir_line_from_graph(GRAPH* pGraph, MINI_GENERATE_TREE* pMiniTree, DIR_LINE* pDirLine)
{
	DIR_LINE* pHead;
	DIR_LINE* prev;
	VECTEX* pVectex;
	LINE* pLine;
	int index;
	int start;

	pHead = NULL;
	for(index = 0; index < pMiniTree->node_num; index++){
		start = pMiniTree->pNode[index];
		pVectex = find_vectex_in_graph(pGraph->head, start);
		pLine = pVectex->neighbor;

		while(pLine){
			insert_line_into_queue(&pHead, start, pLine->end, pLine->weight);
			pLine = pLine->next;
		}
	}

	if(NULL == pHead)
		return;

	delete_unvalid_line_from_list(&pHead, pMiniTree);
	if(NULL == pHead)
		return;

	sort_for_line_list(&pHead);
	memmove(pDirLine, pHead, sizeof(DIR_LINE));

	while(pHead){
		prev = pHead;
		pHead = pHead->next;
		free(prev);
	}
	return;
}


    e)添加节点函数,将尚不是最小生成树的点纳入到最小生成树当中去

void insert_node_into_mini_tree(DIR_LINE* pLine, MINI_GENERATE_TREE* pMiniTree)
{
	int index;

	for(index = 0; index < pMiniTree->node_num; index ++){
		if(pLine->start == pMiniTree->pNode[index]){
			pMiniTree->pNode[pMiniTree->node_num++] = pLine->end;
			return;
		}
	}

	pMiniTree->pNode[pMiniTree->node_num++] = pLine->start;
	return;
}

注意事项:

    (1)d、e是c中调用的子函数,如果大家观察一下就明白了

    (2)最小生成树是按照自顶向下的顺序编写的,虽然c中的子函数完成了,但是d中还有两个子函数没有着落

    (3)d中的函数delete_unvalid_line_from_list、sort_for_line_list会在下一篇中继续介绍

    (4)算法只要能够按照手工计算的流程编写出来,基本上问题不大,但是一些细节还是要小心注意的


【待续】


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

标签:style   blog   http   ar   color   sp   for   strong   on   

原文地址:http://www.cnblogs.com/lonelyxmas/p/4156952.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!