在树之前的笔记中讲到,树这种结构对比数组,链表,增删改查都很快。
二叉排序树简介
二叉排序树:BST,对于二叉排序树的任何一个非叶子节点要求左子节点的值比当前节点的值小,右子节点的值比当前节点的值大。如果有相同的值,可以放在左右子节点都可以,但是要满足前面的条件。
思路
添加
- 传入一个节点,这个节点得值和当前节点的值进行比较
- 如果传入节点的值比当前节点得值小
- 如果当前节点的左子节点为null,就把传入节点赋值给当前节点的左子节点
- 否则就向当前节点的左子节点递归添加
第2步否则(就是传入的节点的值大于等于当前节点的值)
删除叶子节点
- 找到要删除节点target和它的父节点parent
- 确定target是parent的左右哪个子节点
- 根据前面的情况
parent.left=null;
或者parent.right=null;
删除只有一棵子树的节点
- 找到要删除节点target和它的父节点parent
- 确认target是parent的左右子节点
- 确定target的子树是它的左右子节点
- 然后target的对应子树的指针,赋值给parent对应taeget的哪个指针
删除有两棵子树的节点
- 找到要删除节点target和它的父节点parent
- 从target的右子树找到最小的节点(从左子树找,就要找到左子树最大的值,下面步骤同理)
- 用一个临时变量,将这个最小节点的值保存
- 删除该最小节点
- 这个临时变量赋值给target的值
代码
二叉排序树中序遍历出来就是拍好的顺序
using System;
using System.Collections.Generic;
using System.Text;
namespace ConsoleApp1
{
public class BST
{
public Node root;
//添加
public void add (Node node)
{
if (root == null)
root = node;
else
root.add(node);
}
//中序遍历
public void infixOrder()
{
if (root == null)
{
Console.WriteLine("二叉树为空");
return;
}
root.infixOrder();
}
//查找要删除的节点
public Node search(int val)
{
if (root == null)
return null;
else
return root.search(val);
}
//查找要删除节点的父节点
public Node searchParent(int val)
{
if (root == null)
return null;
else
return root.searchParent(val);
}
//删除右边节点的最小的一个节点
//返回以node为根节点的二叉排序树的最小节点的值,并且删除这个节点
public int delRightTreeMin(Node node)
{
Node target = node;
while (target.left != null)
{
target = target.left;
}
//这是target就指向了最小的节点
//删除这个最小节点,并且返回这个节点的值
delNode(target.id);
return target.id;
}
//删除节点
public void delNode(int val)
{
if (root == null)
return;
else
{
//找要删除的节点target和它的父节点parent
Node target = search(val);
if (target == null)//都没有找到要删除的节点,下面的代码也不用执行了
return;
//如果这个二叉树只有一个节点
if (root.left == null && root.right == null)
{
root = null;
return;
}
Node parent = searchParent(val);
//要删除的节点是叶子节点
if (target.left == null && target.right == null)
{
if (parent.left == target)
parent.left = null;
if (parent.right == target)
parent.right = null;
}
//如果要删除的节点有两棵子树
else if (target.left != null && target.right != null)
{
int minval = delRightTreeMin(target.right);
target.id = minval;
}
//要删除的节点只有一棵子树
else
{
if(target.left!=null)//要删除的节点有左子树
{
if (parent != null)
{
if (parent.left == target)
parent.left = target.left;
if (parent.right == target)
parent.right = target.left;
}
else
root = target.left;
}
else//要删除的节点有右子树
{
if (parent != null)
{
if (parent.left == target)
parent.left = target.right;
if (parent.right == target)
parent.right = target.right;
}
else
root = target.right;
}
}
}
}
}
public class Node
{
public int id;
public Node left;
public Node right;
public Node(int id)
{
this.id = id;
}
//二叉排序树添加
public void add(Node node)
{
if (node == null)
return;
//判断传入的节点的值,和当前子树的根节点的关系
if (node.id < this.id)
{
//如果当前的左子节点为null
if (this.left == null)
this.left = node;
else
this.left.add(node);//递归向左子节点添加
}
else
{
if (this.right == null)
this.right = node;
else
this.right.add(node);//递归向右节点添加
}
}
//中序遍历
public void infixOrder()
{
//1.如果当前节点的左子节点不为空,则递归继续中序遍历
if (this.left != null)
this.left.infixOrder();
//2.输出当前节点
Console.WriteLine(this.id);
//3.如果当前节点的右子节点不为空,则递归继续中序遍历
if (this.right != null)
this.right.infixOrder();
}
//查找节点
public Node search(int val)
{
if (val == this.id)
return this;
else if (val < this.id)
{
if (this.left == null)
return null;
return this.left.search(val);
}
else
{
if (this.right == null)
return null;
return this.right.search(val);
}
}
//查找要删除节点的父节点
public Node searchParent(int val)
{
//如果当前节点就是要删除的节点的父节点,就返回
if (this.left != null && this.left.id == val || this.right != null && this.right.id == val)
return this;
else
{
if (val < this.id && this.left != null)//左递归
return this.left.searchParent(val);
else if (val >= this.id && this.right != null)//右递归
return this.right.searchParent(val);
else//没有父节点
return null;
}
}
}
}