二叉搜索树---AVL树删除节点

前面我们写了AVL树的创建AVL树插入节点,即节点的插入,下面我们介绍AVL树节点的删除,与前面调整的 方法相同,在树不平衡时,对树进行调整,具体步骤如下:

1.
- 判断树是否为NULL,若为NULL,直接返回false;
- 判断树是否只有一个节点,且该节点为要删除的节点,直接删除,_pRoot置NULL;
- 否则的话找到要删除节点的位值,删除该节点(分三种情况),并用其孩子节点来代替要删除的节点,该过程树可能出现不平衡,需要调整;

2.删除节点:
- 当要删除的节点只有左孩子;
- 要删除的节点只有右孩子;
- 要删除的节点左右孩子都存在,此时中序遍历其右子树的第一个节点,替换并删除;

3.删除后更新平衡因子,判断树是否平衡并做相应的旋转处理

代码实现:

    bool _Remove(const k& key,const v& value)
    {
        if(_pRoot == NULL)
            return false;
        if(_pRoot->_pLeft == NULL && _pRoot->_pRight == NULL && _pRoot->_key == key)
        {
            delete _pRoot;
            _pRoot = NULL;
        }

        //查找要删除的位置
        Node* pNode = _pRoot;
        Node* pParent = NULL;

        while(pNode)
        {
            if(pNode->_key > key)
            {
                pParent = pNode;
                pNode = pNode->_pLeft;
            }
            else if(pNode->_key < key)
            {
                pParent = pNode;
                pNode = pNode->_pRight;
            }
            else
                break;
        }
        //已找到要删除的节点
        Node* pDel = NULL;

        if(pNode)
        {
            if(pNode->_pLeft == NULL)//只有右子树存在
            {
                if(_pRoot->_key == pNode->_key)
                {
                    pDel = pNode;
                    _pRoot = pNode->_pRight;
                    delete pDel;
                    //删除根节点不需要调整
                    return true;
                }
                else
                {
                    if(pParent->_pLeft == pNode)
                    {
                        pDel = pNode;
                        pParent->_pLeft = pNode->_pRight;
                        pNode = pParent->_pLeft;
                        //更新平衡因子
                        //pParent->_bf++;
                    }
                    else
                    {
                        pDel = pNode;
                        pParent->_pRight = pNode->_pRight;
                        pNode = pParent->_pRight;
                    }
                }
            }
            else if(pNode->_pRight == NULL)//只有左子树
            {
                if(_pRoot == pNode)
                {
                    pDel = pNode;
                    _pRoot = pNode->_pLeft;
                    delete pDel;
                    //删除根节点不需要调整
                    return true;
                }
                else
                {
                    if(pParent->_pLeft = pNode)
                    {
                        pDel = pNode;
                        pParent->_pLeft = pNode->_pLeft;
                        pNode = pParent->_pLeft;
                    }
                    else
                    {
                        pDel = pNode;
                        pParent->_pRight = pNode->_pLeft;
                        pNode = pParent->_pRight;
                    }
                }
            }
            else //左右子树都存在
            {
                Node* pCur = pNode;
                while(pCur->_pLeft)
                {
                    pParent = pCur;
                    pCur = pCur->_pLeft;//找到中序遍历的第一个节点
                }

                pNode->_key = pCur->_key;
                pNode->_value = pCur->_value;


                pDel = pCur;//????

                pParent->_pLeft = pCur->_pRight;
                pNode = pParent->_pLeft;

            }
            delete pDel;
        }


        //删除了节点pDel,更新平衡因子
        while(pParent)
        {
            if(pParent->_pLeft == pNode)//删除的是左孩子
                pParent->_bf++;
            else if(pParent->_pRight == pNode)//删除的是右孩子
                pParent->_bf--;
            if(pParent->_bf == 1 || pParent->_bf == -1)//pParent左右孩子都存在,删除节点不影响他的高度
                return true;
            else if(pParent->_bf == 0)
            {
                //if(pParent->_pLeft == pDel)//pParent存在左孩子
                //{}
                pNode = pParent;
                pParent = pParent->_pParent;
            }
            else
            {
                //不满足平衡树,要做旋转处理
                if(pParent->_bf == 2)//右子树
                {
                    if(pNode==NULL ||pNode->_bf == 1)//右侧
                        _RotateL(pParent);//左旋调整
                    else//左侧
                        _RotateRL(pParent);//先右旋再左旋
                }
                else//左子树
                {
                    if(pNode==NULL || pNode->_bf == -1)//左侧
                        _RotateR(pParent);
                    else//右侧
                        _RotateLR(pParent);
                }
                break;//调整完后跳出循环
            }   
        }
    }

旋转处理见上篇博客,AVL的创建,这里不加赘述;

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值