BST的插入+删除任意节点的递归实现

主要是在bst上删除任意一个节点。

通过寻找中序后继节点来完成连接。

 

#include <iostream>
#include <vector>
#include <cstdio>
#include <queue>
#include <cmath>
#include <string.h>
using namespace std;
const int maxn = 2e5+32;
#define LL long long
const int inf = 0x3f3f3f3f;



class bst
{
public:
	struct node
	{
		int key;
		node *left;
		node *right;
		node(int k):key(k),left(nullptr),right(nullptr){}
	};

	node *head=nullptr;

	void insert(node *now,int key)
	{
		if(head==nullptr){
			node *temp=new node(key);
			head=temp;
			return;
		}
		if(key>now->key){
			if(now->right==nullptr){
				node *temp=new node(key);
				now->right=temp;
				return;
			}
			insert(now->right,key);
		}
		else if(key==now->key)
			return;
		else{
			if(now->left==nullptr){
				node *temp=new node(key);
				now->left=temp;
				return;
			}
			insert(now->left,key);
		}
	}
	void midout(node *now)//中序输出
	{
		if(now!=nullptr){
			midout(now->left);
			printf("%d\n",now->key);
			midout(now->right);
		}
	}
	node* findmin(node *now)
	{
		return now->left==nullptr?now:findmin(now->left);
	}
	node* deletemin(node *now)
	{
		if(now->left==nullptr){
			node *temp=now->right;
			delete now;
			return temp;
		}
		now->left=deletemin(now->left);
		return now;
	}
	node* deletekey(node *now,int key)
	{
		if(now->key==key){
			node* temp=findmin(now->right);//找到中序后继节点temp（右子树里的最小值）
			temp->right=deletemin(now->right);//把右子树里的最小值删除，返回的是这棵右子树，把它赋值给中序后继节点temp的right。
			temp->left=now->left;//中序后继节点的left就是被删节点的left
			return temp;
		}else if(now->key<key){
			now->right=deletekey(now->right,key);
			return now;
		}else{
			now->left=deletekey(now->left,key);
			return now;
		}
	}





};


int main()
{
	bst a;
	a.insert(a.head,10);
	a.insert(a.head,15);
	a.insert(a.head,7);
	a.insert(a.head,4);
	a.insert(a.head,8);
	a.midout(a.head);
	cout<<" ****\n";
	a.head=a.deletekey(a.head,10);//把头节点删除了，必须更新一下头节点
	a.midout(a.head);




}