二叉搜索树——插入 | Binary Search Tree 1 | C/C++实现

问题描述

下述伪代码所示的insert用于在二叉搜索树T中插入新值v。insert将键值为v，左子树为NIL，右子树为NIL的点z作为实参，插入T的适当位置。

insert(T, z)
	y = NIL //x的父结点
	x = 'T的根结点'
	while x != NIL
		y = x//设置父结点
		if z.key < x.key
			x = x.left//移动至左子结点
		else
			x = x.right//移动至右子结点
	z.p = y

	if y == NIL//T为空时
		'T的根结点' = z
	else if z.key < y.key
		y.left = z//将z定为y的左子结点
	else
		y.right = z//将z定为y的右子结点

请编写一个程序，对二叉搜索树T执行下述指令：

insert k：在T中插入键值k
print：分别用树的中序遍历和前序遍历算法输出键值

请按照上述伪代码实现插入算法。

输入： 第1行输入命令数m。接下来m行以insert k或者print的格式输入命令，每个命令占1行。
输出： 每执行1次print命令后，就分别输出中序遍历算法和前序遍历算法所得的键值序列，每个序列占一行。每个键值前输出1个空格。
限制：
命令数不超过500000。另外，print命令不超过10个。
-2000000000 ≤ k ≤ 2000000000
采用上述伪代码所示的算法时，树高不可超过100。
二叉搜索树中各结点的键值不重复。

输入示例

8
insert 30
insert 88
insert 12
insert 1
insert 20
insert 17
insert 25
print

输出示例

1 12 17 20 25 30 88
30 12 1 20 17 25 88

讲解

二叉树的搜索结点

struct Node {
	int key;
	Node *parent, *left, *right;
};

结构体Node包含4个成员，分别为键值key、指向父结点的指针parent、指向左子结点的指针left、指向右子结点的指针*right。

insert操作需要在保证二叉搜索树性质的前提下将给定数据插入恰当位置。

insert以根为起点寻找结点z该插入的位置。设当前结点为x，如果z的键值小于x则将当前结点的左子结点作为下一个x，反之则以右子结点作为下一个x，如此不断向叶结点搜索。在此过程中，程序将前一个结点保存在y里，用作z的候选父结点。当x抵达NIL时搜索结束，此时的y就是z的父结点了。

如果搜索结束时y仍为初始值NIL，就代表插入前的二叉树为空，z即成为根结点。如果插入前的二叉树不为空，那么插入的结点z将根据键值成为y左子结点或右子结点。

AC代码如下

#include<cstdio>
#include<cstdlib>
#include<string>
#include<iostream>
using namespace std;

struct Node{
	int key;
	Node *right, *left, *parent;
}; 

Node *root, *NIL;

void insert(int k){
	Node *y = NIL;
	Node *x = root;
	Node *z;
	
	z = (Node *)malloc(sizeof(Node));
	z->key = k;
	z->left = NIL;
	z->right = NIL;
	
	while(x != NIL){
		y = x;
		if(z->key < x->key){
			x = x->left;
		} else {
			x = x->right;
		}
	}
	
	z->parent = y;
	if(y == NIL){
		root = z;
	} else {
		if(z->key < y->key){
			y->left = z;
		} else {
			y->right = z;
		}
	}
}

void inorder(Node *u){
	if(u == NIL) return;
	inorder(u->left);
	printf(" %d", u->key);
	inorder(u->right);
}

void preorder(Node *u){
	if(u == NIL) return;
	printf(" %d", u->key);
	preorder(u->left);
	preorder(u->right);
}

int main(){
	int n, i, x;
	string com;
	
	scanf("%d", &n);
	
	for(i = 0; i < n; i++){
		cin>>com;
		if(com == "insert"){
			scanf("%d", &x);
			insert(x);
		} else if (com == "print"){
			inorder(root);
			printf("\n");
			preorder(root);
			printf("\n");
		}
	}
	
	return 0;
}