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

2014-11-23 23:36:32 · 作者: · 浏览: 32
pTreeNode, 10));

assert(8 == pTreeNode->data);

assert(NULL == pTreeNode->parent);

assert(NULL == pTreeNode->left_child->right_child);

assert(NULL == pTreeNode->parent);

free(pTreeNode->left_child);

free(pTreeNode->right_child);

free(pTreeNode);

} 至此,删除节点为根节点的情形全部讨论完毕,那么如果删除的节点是普通节点呢,那应该怎么解决呢?

STATUS delete_node_from_tree(TREE_NODE** ppTreeNode, int data)

{

TREE_NODE* pTreeNode;

TREE_NODE* pLeftMax;

if(NULL == ppTreeNode || NULL == *ppTreeNode)

return FALSE;

pTreeNode = find_data_in_tree_node(*ppTreeNode, data);

if(NULL == pTreeNode)

return FALSE;

if(*ppTreeNode == pTreeNode){

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

*ppTreeNode = NULL;

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

*ppTreeNode = pTreeNode->left_child;

pTreeNode->left_child->parent = NULL;

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

*ppTreeNode = pTreeNode->right_child;

pTreeNode->right_child->parent = NULL;

}else{

pLeftMax = find_max_node(pTreeNode->left_child);

if(pLeftMax == pTreeNode->left_child){

*ppTreeNode = pTreeNode->left_child;

(*ppTreeNode)->right_child = pTreeNode->right_child;

(*ppTreeNode)->right_child->parent = *ppTreeNode;

(*ppTreeNode)->parent = NULL;

}else{

pTreeNode->data = pLeftMax->data;

pLeftMax->parent->right_child = NULL;

pTreeNode = pLeftMax;

}

}

free(pTreeNode);

return TRUE;

}

return _delete_node_from_tree(pTreeNode);

}

STATUS delete_node_from_tree(TREE_NODE** ppTreeNode, int data)

{

TREE_NODE* pTreeNode;

TREE_NODE* pLeftMax;

if(NULL == ppTreeNode || NULL == *ppTreeNode)

return FALSE;

pTreeNode = find_data_in_tree_node(*ppTreeNode, data);

if(NULL == pTreeNode)

return FALSE;

if(*ppTreeNode == pTreeNode){

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

*ppTreeNode = NULL;

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

*ppTreeNode = pTreeNode->left_child;

pTreeNode->left_child->parent = NULL;

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

*ppTreeNode = pTreeNode->right_child;

pTreeNode->right_child->parent = NULL;

}else{

pLeftMax = find_max_node(pTreeNode->left_child);

if(pLeftMax == pTreeNode->left_child){

*ppTreeNode = pTreeNode->left_child;

(*ppTreeNode)->right_child = pTreeNode->right_child;

(*ppTreeNode)->right_child->parent = *ppTreeNode;