二叉搜索树BST(Binary Search Tree)的创建

以下是鄙人创建此树的全部过程

1. 生成结构体

树节点中包含1个数值域,两个指针域,分别指向左右孩子。
结构示意:
在这里插入图片描述

定义如下:

struct TreeNode {
    
    
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode() :val(0), left(nullptr), right(nullptr) {
    
    }
	TreeNode(int x) :val(x), left(nullptr), right(nullptr) {
    
    }
	TreeNode(int x, TreeNode* left, TreeNode* right) :val(x), left(left), right(right) {
    
    }
};

2. 类的定义

BST中封装了root变量,使用getroot成员函数可以获取root,声明了函数buildBST

class BST {
    
    
	TreeNode* root;
public:
	TreeNode* getroot()
	 {
    
    
		return root;
	}
	void buildBST(vector<int>& vec);
};

3. 搜索树的构建

void  BST::buildBST(vector<int>& vec) {
    
    
	if (vec.size() == 0)
		root = nullptr;//若为空树,则置空

	root = new TreeNode(vec[0]);//创建根节点,并初始化为vec[0]
	for (int i = 1; i < vec.size(); i++) {
    
    
		TreeNode* newNode = new TreeNode(vec[i]);//创建节点
		TreeNode* Root = root;//保存当前根节点,用Root寻找插入位置
		while (Root) {
    
    
			if (newNode->val < Root->val) {
    
    //若小于根节点,放在节点左侧
				if (Root->left == nullptr) {
    
    //当根节点没有左孩子时,放置到此,此次插入结束,跳出循环
					Root->left = newNode;
					break;
				}
					Root = Root->left;//继续深入寻找
			}
			else {
    
    
				if (Root->right == nullptr) {
    
    
					Root->right = newNode;
					break;
				}
				else {
    
    
					Root = Root->right;
				}
			}
		}
	}
}

4. 中序打印输出搜索树

根据搜索树的特点,中序打印其就能得到一串从小到大的数列。
使用的是递归法中序遍历:

void InOrderprint(TreeNode* root) {
    
    //中序打印树
	if (root == nullptr)
		return;
	InOrderprint(root->left);
	cout << root->val << " ";
	InOrderprint(root->right);
}

完整代码:

#include <iostream>
#include <vector>
using namespace std;

struct TreeNode {
    
    
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode() :val(0), left(nullptr), right(nullptr) {
    
    }
	TreeNode(int x) :val(x), left(nullptr), right(nullptr) {
    
    }
	TreeNode(int x, TreeNode* left, TreeNode* right) :val(x), left(left), right(right) {
    
    }
};

class BST {
    
    
	TreeNode* root;
public:
	TreeNode* getroot() {
    
    
		return root;
	}
	void buildBST(vector<int>& vec);
};

void  BST::buildBST(vector<int>& vec) {
    
    
	if (vec.size() == 0)
		root = nullptr;

	root = new TreeNode(vec[0]);//创建根节点,并初始化为vec[0]
	for (int i = 1; i < vec.size(); i++) {
    
    
		TreeNode* newNode = new TreeNode(vec[i]);//创建节点
		TreeNode* Root = root;//保存当前根节点,用Root寻找插入位置
		while (Root) {
    
    
			if (newNode->val < Root->val) {
    
    //若小于根节点,放在节点左侧
				if (Root->left == nullptr) {
    
    //当根节点没有左孩子时,放置到此
					Root->left = newNode;
					break;
				}
					Root = Root->left;//继续深入寻找
			}
			else {
    
    
				if (Root->right == nullptr) {
    
    
					Root->right = newNode;
					break;
				}
				else {
    
    
					Root = Root->right;
				}
			}
		}
	}
}

void InOrderprint(TreeNode* root) {
    
    //中序打印树
	if (root == nullptr)
		return;
	InOrderprint(root->left);
	cout << root->val << " ";
	InOrderprint(root->right);
}


int main(int argc, char* argv[])
{
    
    
	vector<int> vec = {
    
     8,4,3,2,1,5,6,7 };
	BST bst;
	bst.buildBST(vec);
	TreeNode* roo = bst.getroot();
	InOrderprint(roo);
	return 0;
}

运行效果图:

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/Genius_bin/article/details/113176739