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

2014-11-23 23:36:33 · 作者: · 浏览: 49
20

*/

static void test13()

{

TREE_NODE* pTreeNode = NULL;

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

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

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

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

assert(20 == pTreeNode->right_child->data);

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

free(pTreeNode->right_child);

free(pTreeNode);

}

/*

*

* 10 ======> 10

* \ \

* 15 20

* \

* 20

*/

static void test13()

{

TREE_NODE* pTreeNode = NULL;

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

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

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

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

assert(20 == pTreeNode->right_child->data);

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

free(pTreeNode->right_child);

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;

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

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

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

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

else

pTreeNode->parent->right_child = pTreeNode->right_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;

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