[保研/考研机试] KY223 二叉排序树华中科技大学复试上机题 C++实现

news2024/11/22 11:12:51

题目链接：

二叉排序树_牛客题霸_牛客网输入一系列整数，建立二叉排序树，并进行前序，中序，后序遍历。。题目来自【牛客题霸】https://www.nowcoder.com/share/jump/437195121692722441741

描述

输入一系列整数，建立二叉排序树，并进行前序，中序，后序遍历。

输入描述：

输入第一行包括一个整数n(1<=n<=100)。接下来的一行包括n个整数。

输出描述：

可能有多组测试数据，对于每组数据，将题目所给数据建立一个二叉排序树，并对二叉排序树进行前序、中序和后序遍历。每种遍历结果输出一行。每行最后一个数据之后有一个换行。输入中可能有重复元素，但是输出的二叉树遍历序列中重复元素不用输出。

示例1：

输入：

5
1 6 5 9 8

输出：

1 6 5 9 8 
1 5 6 8 9 
5 8 9 6 1

思路：

定义了二叉排序树节点结构 TreeNode，包含数据域 data，左子节点指针 leftChild，右子节点指针 rightChild。
Insert 函数用于将元素插入到二叉排序树中。根据比较大小关系，元素被插入到左子树或右子树中。
PreOrder 函数实现了先序遍历，遍历顺序为：根节点 -> 左子树 -> 右子树，同时输出节点数据。
InOrder 函数实现了中序遍历，遍历顺序为：左子树 -> 根节点 -> 右子树，同时输出节点数据。
PostOrder 函数实现了后序遍历，遍历顺序为：左子树 -> 右子树 -> 根节点，同时输出节点数据。
在 main 函数中，首先读取一个整数 n，表示要插入的节点个数。然后使用循环读取 n 个节点值，并调用 Insert 函数将节点插入到二叉排序树中。最后分别调用 PreOrder、InOrder 和 PostOrder 函数进行不同遍历方式，并输出遍历结果。

该代码主要实现了二叉排序树的构建以及三种不同的遍历方式：先序遍历、中序遍历和后序遍历，并输出遍历结果。

源代码：

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

// 定义二叉排序树节点结构
struct TreeNode {
	int data; // 节点数据
	TreeNode* leftChild; // 左子节点指针
	TreeNode* rightChild; // 右子节点指针
	TreeNode(int x) : data(x), leftChild(NULL), rightChild(NULL) {} // 构造函数
};

// 插入元素到二叉排序树中
TreeNode* Insert(TreeNode* root, int x) {
	if (root == NULL) {
		root = new TreeNode(x); // 创建新节点
	}
	else if (x < root->data) {
		root->leftChild = Insert(root->leftChild, x); // 向左子树插入
	}
	else if (x > root->data) {
		root->rightChild = Insert(root->rightChild, x); // 向右子树插入
	}
	return root;
}

// 先序遍历
void PreOrder(TreeNode* root) {
	if (root == NULL) {
		return;
	}
	cout << root->data << " "; // 输出节点数据
	PreOrder(root->leftChild); // 递归遍历左子树
	PreOrder(root->rightChild); // 递归遍历右子树
}

// 中序遍历
void InOrder(TreeNode* root) {
	if (root == NULL) {
		return;
	}
	InOrder(root->leftChild); // 递归遍历左子树
	cout << root->data << " "; // 输出节点数据
	InOrder(root->rightChild); // 递归遍历右子树
}

// 后序遍历
void PostOrder(TreeNode* root) {
	if (root == NULL) {
		return;
	}
	PostOrder(root->leftChild); // 递归遍历左子树
	PostOrder(root->rightChild); // 递归遍历右子树
	cout << root->data << " "; // 输出节点数据
}

int main() {
	int n;
	while (cin >> n) {
		TreeNode* root = NULL; // 树的根节点
		for (int i = 0; i < n; i++) {
			int temp;
			cin >> temp; // 读取输入的节点值
			root = Insert(root, temp); // 插入节点到二叉排序树中
		}
		PreOrder(root); // 先序遍历并输出结果
		cout << endl;
		InOrder(root); // 中序遍历并输出结果
		cout << endl;
		PostOrder(root); // 后序遍历并输出结果
		cout << endl;
	}

	return 0;
}