1043 Is It a Binary Search Tree (25)

最新推荐文章于 2024-11-27 20:25:31 发布

wangfy_

最新推荐文章于 2024-11-27 20:25:31 发布

阅读量237

点赞数

CC 4.0 BY-SA版权

分类专栏：二叉搜索树 pat-a

本文链接：https://blog.youkuaiyun.com/chushoufengli/article/details/79748764

pat-a 同时被 2 个专栏收录

35 篇文章

订阅专栏

二叉搜索树

5 篇文章

订阅专栏

本文介绍了一种算法，该算法接收一系列整数并判断这些整数是否能构成一棵二叉搜索树（BST）或其镜像树的先序遍历序列。若条件满足，则输出YES及相应的后序遍历序列；否则输出NO。通过递归方式实现节点插入、遍历功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//给定N个整数作为BST的结点插入顺序，如果这棵BST或镜像BST的先序等于输入顺序，则输出YES，并输出相应BST的后序，否则输出NO
//code:
#include <cstdio>
#include <vector>
using namespace std;
int N;
vector<int> origin;
vector<int> preOrder_vector;
vector<int> preOrder_mirror_vector;
int j;
struct node {
	int data;
	node* lChild;
	node* rChild;
};
void insert(node* &root, int data) {  //二叉树的每次插入，都先从root开始
	if(root == NULL) {
		root = new node; 
		root->data = data;
		root->lChild = root->rChild = NULL;
		return;
	}
	if(data >= root->data) insert(root->rChild, data);  //右子树 >= 根结点
	else insert(root->lChild, data);
}
void insert_mirror(node* &root,int data) {
	if(root == NULL) {
		root = new node;
		root->data = data;
		root->lChild = root->rChild = NULL;
		return;
	}
	if(data >= root->data) insert_mirror(root->lChild, data);
	else insert_mirror(root->rChild, data);
}
void preOrder(node* root, vector<int>& vi) {
	if(root == NULL) return;
	vi.push_back(root->data);
	preOrder(root->lChild, vi);
	preOrder(root->rChild, vi);
}
void preOrder_mirror(node* root, vector<int>& vi) {
	if(root == NULL) return;
	vi.push_back(root->data);
	preOrder_mirror(root->lChild, vi);
	preOrder_mirror(root->rChild, vi);
}
void postOrder(node* root) {
	if(root == NULL) return;
	postOrder(root->lChild);
	postOrder(root->rChild);
	if(j != N) {
		printf("%d ",root->data);
		j ++;
	} else {
		printf("%d",root->data);
	}
}
int main() {
	node* root = NULL;  ////
	node* root_mirror = NULL;
	scanf("%d",&N);
	for(int i=0; i<N; i++) {
		int c;
		scanf("%d",&c);
		origin.push_back(c);
		insert(root, c);
		insert_mirror(root_mirror, c);
	}
	preOrder(root, preOrder_vector);
	preOrder_mirror(root_mirror, preOrder_mirror_vector);
	if(preOrder_vector == origin) {  ////vector可以直接比较
		printf("YES\n");
		j = 1;  //用于控制输出
		postOrder(root);
	} else if(preOrder_mirror_vector == origin) {
		printf("YES\n");
		j = 1;
		postOrder(root_mirror);
	} else {
		printf("NO");
	}
	return 0;
}