一步一步写算法(之排序二叉树插入)

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


    二叉树的节点插入比较简单。一般来说,二叉树的插入主要分为以下两个步骤:

    1) 对当前的参数进行判断,因为需要考虑到头结点,所以我们使用了指针的指针作为函数的输入参数

    2) 分情况讨论:

        如果原来二叉树连根节点都没有,那么这个新插入的数据就是根节点;

        如果原来的二叉树有根节点,那我们判断这个数据是否存在过,如果存在,那么返回;如果不存在,那么继续插入数据。

        那继续插入的数据怎么保存呢?又要分三种情况:

             1)如果插入的数据小于当前节点的数据,那么往当前节点的左子树方向继续寻找插入位置

             2)如果插入的数据大于当前插入的位置,那么往当前节点的右子树方向继续寻找插入位置

             3)如果方向当前的节点为空,那么表示插入的位置找到了,插入数据即可

    算法说了这么多,下面即开始练习我们的代码:

    a)判断输入数据的合法性

STATUS insert_node_into_tree(TREE_NODE** ppTreeNode, int data)
{
	if(NULL == ppTreeNode)
		return FALSE;

	return TRUE;
}
    此时,可以用一个测试用例验证一下

static void test1()
{
	assert(FALSE == insert_node_into_tree(NULL, 10));
}

     b)判断当前根节点是否存在,修改代码

STATUS insert_node_into_tree(TREE_NODE** ppTreeNode, int data)
{
	if(NULL == ppTreeNode)
		return FALSE;

	if(NULL == *ppTreeNode){
		*ppTreeNode = (TREE_NODE*)create_tree_node(data);
		assert(NULL != *ppTreeNode);
		return TRUE;
	}

	return TRUE;
}
    修改了代码,少不了测试用例的添加。

static void test2()
{
	TREE_NODE* pTreeNode = NULL;
	assert(TRUE == insert_node_into_tree(&pTreeNode, 10));
	assert(10 == pTreeNode->data);
	free(pTreeNode);
}
  

    c)上面考虑了没有根节点的情况,那么如果根节点存在呢?

STATUS _insert_node_into_tree(TREE_NODE** ppTreeNode, int data, TREE_NODE* pParent)
{
	if(NULL == *ppTreeNode){
		*ppTreeNode = create_tree_node(data);
		assert(NULL != *ppTreeNode);
		(*ppTreeNode)->parent = pParent;
		return TRUE;
	}

	if(data < (*ppTreeNode)->data)
		return _insert_node_into_tree(&(*ppTreeNode)->left_child, data, *ppTreeNode);
	else
		return _insert_node_into_tree(&(*ppTreeNode)->right_child, data, *ppTreeNode);
}

STATUS insert_node_into_tree(TREE_NODE** ppTreeNode, int data)
{
	if(NULL == ppTreeNode)
		return FALSE;

	if(NULL == *ppTreeNode){
		*ppTreeNode = (TREE_NODE*)create_tree_node(data);
		assert(NULL != *ppTreeNode);
		return TRUE;
	}

	return _insert_node_into_tree(ppTreeNode, data, NULL);
}
    上面的代码已经考虑了不是根节点的情况。我们可以据此添加一个测试用例。

static void test3()
{
	TREE_NODE* pTreeNode = NULL;
	assert(TRUE == insert_node_into_tree(&pTreeNode, 9));
	assert(TRUE == insert_node_into_tree(&pTreeNode, 8));
	assert(TRUE == insert_node_into_tree(&pTreeNode, 10));
	assert(9 == pTreeNode->data);
	assert(8 == pTreeNode->left_child->data);
	assert(10 == pTreeNode->right_child->data);
	free(pTreeNode->left_child);
	free(pTreeNode->right_child);
	free(pTreeNode);
}
    由于上面的代码是递归代码,为了实现代码的健壮性和完毕性,其实我们设计测试用例的时候应该至少包括9个测试用例:

    (1) 参数非法

    (2) 根节点不存在

    (3)根节点存在,但是插入的数据已经存在

    (4)根节点存在,插入数据为 9, 8

    (5)根节点存在, 插入数据为9, 10

    (6)根节点存在,插入数据为9,8, 7

    (7)根节点存在,插入数据为9,7,8

    (8)根节点存在,插入数据为7,8, 9

    (9)根节点存在,插入数据为7,9,8


【预告: 下面一篇博客主要介绍二叉树的节点删除】


  • 14
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 8
    评论
好的,以下是一个基于C语言的二叉树排序算法的代码示例: ``` #include <stdio.h> #include <stdlib.h> // 定义二叉树结构体 typedef struct TreeNode { int value; struct TreeNode *left; struct TreeNode *right; } TreeNode; // 定义二叉树节点的创建函数 TreeNode* create_node(int value) { TreeNode* node = (TreeNode*) malloc(sizeof(TreeNode)); node->value = value; node->left = NULL; node->right = NULL; return node; } // 定义二叉树节点的插入函数 void insert_node(TreeNode **root, int value) { if (*root == NULL) { *root = create_node(value); } else if (value < (*root)->value) { insert_node(&((*root)->left), value); } else { insert_node(&((*root)->right), value); } } // 定义二叉树的前序遍历函数 void preorder_traversal(TreeNode *root) { if (root != NULL) { printf("%d ", root->value); preorder_traversal(root->left); preorder_traversal(root->right); } } int main() { // 创建一个空的二叉树 TreeNode *root = NULL; // 向二叉树插入一些节点 insert_node(&root, 5); insert_node(&root, 3); insert_node(&root, 7); insert_node(&root, 1); insert_node(&root, 9); insert_node(&root, 4); // 输出排序后的结果 printf("排序结果:"); preorder_traversal(root); return 0; } ``` 以上代码中,我们定义了一个二叉树结构体 `TreeNode`,包含了一个 `value` 属性和左右子节点的指针。我们还定义了创建节点函数 `create_node` 和插入节点函数 `insert_node`,以及二叉树的前序遍历函数 `preorder_traversal`。在 `main` 函数中,我们创建一个空的二叉树,并向其中插入一些节点,最后调用前序遍历函数输出排序后的结果。 希望这个代码示例能够对你有所帮助。如果有任何问题或疑惑,欢迎再次向我提问。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

嵌入式-老费

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

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

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

打赏作者

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

抵扣说明:

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

余额充值