一步一步写算法(之排序二叉树删除-3)(四)

2014-11-23 23:36:33 · 作者: · 浏览: 51
ld);

free(pTreeNode);

} 添加左子树不为空,右子树为空的处理代码,如下所示:

STATUS _delete_node_from_tree(TREE_NODE* pTreeNode)

{

TREE_NODE* pLeftMax;

if(NULL == pTreeNode-> left_child && NULL == pTreeNode->right_child){

if(pTreeNode == pTreeNode->parent->left_child)

pTreeNode->parent->left_child = NULL;

else

pTreeNode->parent->right_child = NULL;

}else if(NULL != pTreeNode->left_child && NULL == pTreeNode->right_child){

pTreeNode->left_child->parent = pTreeNode->parent;

if(pTreeNode == pTreeNode->parent->left_child)

pTreeNode->parent->left_child = pTreeNode->left_child;

else

pTreeNode->parent->right_child = pTreeNode->left_child;

}

free(pTreeNode);

return TRUE;

}

STATUS _delete_node_from_tree(TREE_NODE* pTreeNode)

{

TREE_NODE* pLeftMax;

if(NULL == pTreeNode-> left_child && NULL == pTreeNode->right_child){

if(pTreeNode == pTreeNode->parent->left_child)

pTreeNode->parent->left_child = NULL;

else

pTreeNode->parent->right_child = NULL;

}else if(NULL != pTreeNode->left_child && NULL == pTreeNode->right_child){

pTreeNode->left_child->parent = pTreeNode->parent;

if(pTreeNode == pTreeNode->parent->left_child)

pTreeNode->parent->left_child = pTreeNode->left_child;

else

pTreeNode->parent->right_child = pTreeNode->left_child;

}

free(pTreeNode);

return TRUE;

}

3.3 删除的节点左子树为空,右子树节点不为空

测试用例1: 删除数据6

/*

*

* 10 ======> 10

* / /

* 6 8

* \

* 8

*/

static void test12()

{

TREE_NODE* pTreeNode = NULL;

assert(TRUE == insert_node_into_tree(&pTreeNode, 10));

assert(TRUE == insert_node_into_tree(&pTreeNode, 6));

assert(TRUE == insert_node_into_tree(&pTreeNode, 8));

assert(TRUE == delete_node_from_tree(&pTreeNode, 6));

assert(8 == pTreeNode->left_child->data);

assert(pTreeNode = pTreeNode->left_child->parent);

free(pTreeNode->left_child);

free(pTreeNode);

}

/*

*

* 10 ======> 10

* / /

* 6 8

* \

* 8

*/

static void test12()

{

TREE_NODE* pTreeNode = NULL;

assert(TRUE == insert_node_into_tree(&pTreeNode, 10));

assert(TRUE == insert_node_into_tree(&pTreeNode, 6));

assert(TRUE == insert_node_into_tree(&pTreeNode, 8));

assert(TRUE == delete_node_from_tree(&pTreeNode, 6));

assert(8 == pTreeNode->left_child->data);

assert(pTreeNode = pTreeNode->left_child->parent);

free(pTreeNode->left_child);

free(pTreeNode);

} 测试用例2: 删除数据15

/*

*

* 10 ======> 10

* \ \

* 15 20

* \

*