zoukankan      html  css  js  c++  java
  • C++实现二叉搜索树的插入,删除

    二叉搜索树即左孩子的值小于根节点,右孩子的值大于根节点。

    二叉搜索树的插入:

    即检查要插入的数(key,下文都用它表示)是否存在,若存在返回false,不存在即插入,在插入时,若key大于根节点,则插入到右子树中,更新根节点,依次类推;若key小于根节点,则插入到左子树中,更新根节点,以此类推。下面用图表示

    现在要插入13,由二叉树的性质可知,13要插到10的右子树上,即

    代码实现:

    bool Insert(const k& key, const v& value)
     {
      if (_root == NULL)
      {
       _root = new Node(key, value);
      }
      Node* cur = _root;
      Node* parent = cur;
      while (cur)
      {
       if (key > cur->_key)
       {
        parent = cur;
        cur = cur->_right;
       }
       else if (key < cur->_key)
       {
        parent = cur;
        cur = cur->_left;
       }
       else
       {
        return false;
       }
      }
      
      Node* node = new Node(key, value);
      if (key > parent->_key)
      {
       parent->_right = node;
      }
      else
      {
       parent->_left = node;
      }
      return true;

    }
     二叉树的删除:

     二叉树的删除要分三种情况,分别为:

    a、删除的节点为叶子节点

    b、删除的节点有一个子节点

    c、删除的节点有两个子节点

    第一种就不具体说了,直接删除即可。

    下面分析第二种

    如果删除的节点缺左子树就用右子树进行填补,同理,缺右子树就用左子树进行填补。下面用图说明

    要删除1这个节点,即用1 的左子树填补1的位置,再将1删除。如下图

    紧接着第三种情况,也是最复杂的一种。

    要删除的节点有两个子节点,在这里定义删除的节点为del,firstInorder为以del为跟的树的最左节点,交换del和firstInorder的值,删除del即可,下面同样用图展示

    要删除6这个节点,首先找到最左节点7,因为7是右子树中最小的,交换6,7,删除6,此树仍然满足搜索树,如下图

    代码实现如下

    bool Remove(Node* root,const k& key)
     {
      //根节点为空
      if (root == NULL)
      {
       return false;
      }
      //只有一个节点
      if (root->_left == NULL && root->_right == NULL)
      {
       delete root;
       root = NULL;
       return true;
      }

      Node* parent = NULL;
      Node* del = root;
      while (del)
      {
       if (key > del->_key)
       {
        parent = del;
        del = del->_right;
       }
       else if (key < del->_key)
       {
        parent = del;
        del = del->_left;
       }
       else
       {
        break;
       }
      }

      if (del)
      {
       if (del->_left == NULL)
       {
        if (del == root)
        {
         root = del->_right;
        }
        else
        {
         if (del == parent->_left)
         {
          parent->_left = del->_right;
         }
         else
         {
          parent->_right = del->_left;
         }
        } 
       }
       else if (del->_right == NULL)
       {
        if (del = root)
        {
         root = del->_left;
        }
        else
        {
         if (del == parent->_left)
         {
          parent->_left = del->_left;
         }
         else
         {
          parent->_right = del->_right;
         }
        } 
       }
       else
       {
        Node* parent = del;
        Node* firstInorder = del->_right;
        while (firstInorder->_left)
        {
         parent = firstInorder;
         firstInorder = firstInorder->_left;
        }
        swap(del->_key, firstInorder->_key);
        swap(del->_value, firstInorder->_value);    
        parent->_left = firstInorder->_right;
        delete del;
        del = NULL;
        return true;
       }
      } 
     }

    源代码如下:

    template <class k,class v>
    struct BSTNode
    {
     BSTNode<k, v> _left;
     BSTNode<k, v> _right;
     k _key;
     v _value;

     BSTNode(const k& key,const v& value)
      :_left(NULL)
      ,_right(NNULL)
      ,_key(key)
      ,_value(value)
     {}
    };

    template <class k,class v>
    class BSTree
    {
     typedef BSTNode<k, v> Node;
    protected:
     Node* _root;
    public:
     BSTree()
      :_root(NULL)
     {}
    public:
     bool Insert(const k& key, const v& value)
     {
      if (_root == NULL)
      {
       _root = new Node(key, value);
      }
      Node* cur = _root;
      Node* parent = cur;
      while (cur)
      {
       if (key > cur->_key)
       {
        parent = cur;
        cur = cur->_right;
       }
       else if (key < cur->_key)
       {
        parent = cur;
        cur = cur->_left;
       }
       else
       {
        return false;
       }
      }
      
      Node* node = new Node(key, value);
      if (key > parent->_key)
      {
       parent->_right = node;
      }
      else
      {
       parent->_left = node;
      }
      return true;
     }

     bool Insert_R(Node*& _root, const k& key, const v& value)
     {
      if (_root == NULL)
      {
       _root = new Node(key, value);
                return true;
      }
      if (key > _root->_key)
      {
       return Insert_R(_root->_right, key, value);
      }
      else if (key < _root->_key)
      {
       return Insert_R(_root->_left, key, value);
      }
      else
      {
       return false;
      }  
     }

     bool Find(const k& key, const v& value)
     {
      if (_root == NULL)
      {
       return false;
      }
      Node* cur = _root;
      while (cur)
      {
       if (key == cur->_key)
       {
        return true;
       }
       else if (key < cur->_key)
          {
           cur = cur->_right;
          }
       else
       {
        cur = cur->_left;
       }
      }
      return false;
     }

     bool Remove(Node* root,const k& key)
     {
      //根节点为空
      if (root == NULL)
      {
       return false;
      }
      //只有一个节点
      if (root->_left == NULL && root->_right == NULL)
      {
       delete root;
       root = NULL;
       return true;
      }

      Node* parent = NULL;
      Node* del = root;
      while (del)
      {
       if (key > del->_key)
       {
        parent = del;
        del = del->_right;
       }
       else if (key < del->_key)
       {
        parent = del;
        del = del->_left;
       }
       else
       {
        break;
       }
      }

      if (del)
      {
       if (del->_left == NULL)
       {
        if (del == root)
        {
         root = del->_right;
        }
        else
        {
         if (del == parent->_left)
         {
          parent->_left = del->_right;
         }
         else
         {
          parent->_right = del->_left;
         }
        } 
       }
       else if (del->_right == NULL)
       {
        if (del = root)
        {
         root = del->_left;
        }
        else
        {
         if (del == parent->_left)
         {
          parent->_left = del->_left;
         }
         else
         {
          parent->_right = del->_right;
         }
        } 
       }
       else
       {
        Node* parent = del;
        Node* firstInorder = del->_right;
        while (firstInorder->_left)
        {
         parent = firstInorder;
         firstInorder = firstInorder->_left;
        }
        swap(del->_key, firstInorder->_key);
        swap(del->_value, firstInorder->_value);    
        parent->_left = firstInorder->_right;
        delete del;
        del = NULL;
        return true;
       }
      } 
     }

     bool Remove_R(Node*& _root, const k& key, const v& value)
     {
      if (root == NULL)
      {
       return false;
      }
      else if (root->_key < key)
      {
       return _Remove_R(root->_right, key);
      }
      else if (root->_key > key)
      {
       return _Remove_R(root->_left, key);
      }
      else
      {
       Node* del = root;
       if (root->_left == NULL)
       {
        root = root->_right;
       }
       else if (root->_right == NULL)
       {
        root = root->_left;
       }
       else
       {
        Node* firstInorder = root->_right;
        while (firstInorder->_left)
        {
         firstInorder = firstInorder->_left;
        }
        swap(del->_key, firstInorder->_key);
        swap(del->_value, firstInorder->_value);
        return _Remove_R(root->_right, key);
       }
         delete del;
      }
     }

     void _Inorder()
     {
      Inorder(_root);
      cout << endl;
     }

     public:
      Inorder(Node* root)
      {
       if (root == NULL)
       {
        return;
       }
       Inorder(root->_left);
       cout << root->_key << " ";
       Inorder(root->_right);
      }
    };

  • 相关阅读:
    基于Kibana的可视化监控报警插件sentinl入门
    es聚合学习笔记
    spark streaming基本概念一
    es实战一:基本概念
    访问者模式(Visitor Pattern)
    模板方法模式(Template Method Pattern)
    策略模式(Strategy Pattern)
    状态模式(State Pattern)
    观察者(Observer)模式
    备忘录模式(Memento Pattern)
  • 原文地址:https://www.cnblogs.com/tongyan2/p/5459404.html
Copyright © 2011-2022 走看看