二叉搜索树——BST(二叉搜索树)

二叉搜索树——BST(二叉搜索树)

题目：二叉搜索树

题面

题目描述

给定两种操作：

i k：把k加入到集合里面

q k：查询比k小的数有多少个

注：数据保证i操作的k不重复

输入描述

读入一个n，表示有n个操作；

接下来有n行，每行的第一个是字符c，第二个是数字k

输出描述

输出每个q操作的结果，一个一行

输入样例

5
i 3
i 2
i 7
q 5
q 8

输出样例

2
3

数据范围说明

2≤n≤1000

1≤k≤100000

代码及思路

解题思路

这道题要求比k小的数有多少个，这要用到二叉搜索树，我们要先把二叉搜索树构造出来，构造时还要把以这个节点做根节点的树的节点总数记录下来。然后用深度优先搜索来搜索这棵二叉搜索树。先用k与根节点比较，大于根节点，就返回左子树的节点总数+1+继续递归的结果；小于根节点，就继续递归。
思路：

1. 输入
2. 构造二叉搜索树
3. dfs搜索并计数
4. 输出

完整代码

#include<iostream>
using namespace std;
int idx=0;
struct node {
	int date=0;
	int size=0;
	int left=0, right=0;
} a[100005];
void pushup(int xx) {
	a[xx].size=a[a[xx].left].size+a[a[xx].right].size+1;
}
void insert(int x, int k) {
	if(a[x].date==0) {
		idx++;
		a[idx].date=k;
		a[idx].size=1;
		return;
	}
	if(k>=a[x].date && a[x].right==0) {
		++idx;
		a[x].right=idx;
		a[idx].date=k ;
		a[idx].size=1;
		pushup(x);
		return;
	}
	if(k<=a[x]. date && a[x].left==0) {
		++idx;
		a[x].left=idx;
		a[idx]. date=k ;
		a[idx].size=1;
		pushup(x);
		return;
	}
	if(k>=a[x]. date) {
		insert(a[x].right,k);
	} else {
		insert(a[x].left,k);
	}
	pushup(x);
}
int find(int x,int k) {
	if(a[x].date==0) {
		return 0;
	}
	if(a[x].date>=k) {
		return find(a[x].left, k);
	} else {
		return a[a[x].left].size+1+find(a[x].right, k);
	}
}
int main() {
	int n;
	cin>>n;
	for(int i=1; i<=n; i++) {
		char c;
		int s;
		cin>>c>>s;
		if(c=='i') {
			insert (1, s);
		}
		if(c=='q') {
			cout<<find(1,s)<<endl;
		}
	}
	return 0;
}

知识点：BST(二叉搜索树)

性质

二叉查找树（BST），（又称：二叉搜索树，二叉排序树）它或者是一棵空树，或者是具有下列性质的二叉树：

1. 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值；
2. 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值；
3. 它的左、右子树也分别为二叉排序树；
4. 它的中序遍历是升序的。

构造方法

说明

构造二叉搜索树，我们可以从性质入手。根据性质一和性质二，我们可以用深度优先搜索来搜索这棵二叉搜索树。把要插入的节点与根节点比较,
节点大于根节点的并且右子树为空，把节点存进去，右子树不空继续递归；
节点小于根节点的并且左子树为空，把节点存进去，左子树不空继续递归。
这样把所有节点都放进去，二叉搜索树就构造好了。
如果要查找节点也是用这种方式搜索查找的。

代码

下面是我的输入n个节点构造二叉搜索树的程序。

#include<iostream>
using namespace std;
int idx=0;
struct node{
	int data=0;
	int size=0;
	int left=0,right=0;
}a[100005];
void pushup(int xx){
	a[xx].size=a[a[xx].left].size+a[a[xx].right].size+1;
}
void insert(int x,int k){
	if(a[x].data==0){
		idx++;
		a[idx].data=k;
		a[idx].size=1;
		return;
	}
	if(k>=a[x].data && a[x].right==0){
		++idx;
		a[x].right=idx;
		a[idx].data=k;
		a[idx].size=1;
		pushup(k);
		return;
	}
	if(k<=a[x].data && a[x].left==0){
		++idx;
		a[x].left=idx;
		a[idx].data=k;
		a[idx].size=1;
		pushup(k);
		return;
	}
	if(k>=a[x].data){
		insert(a[x].right,k);
	}
	else{
		insert(a[x].left,k);
	}
	pushup(k);
}
int find(int x,int k){
	if(a[x].data==0){
		return 0;
	}
	if(a[x].data>=k){
		return find(a[x].left,k);
	}
	else{
		return a[a[x].left].size+1+find(a[x].right,k);
	}
}
int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		char c;
		int k;
		cin>>c>>k;
		if(c=='i'){
			insert(1,k);
		}
		if(c=='q'){
			cout<<find(1,k)<<endl;
		}
	} 
	return 0;
}

总结

这次我主要讲了以下内容

1. 题目：二叉搜索树
2. 二叉搜索树的性质
3. 二叉搜索树的构造方法