7-8 中序遍历树并判断是否为二叉搜索树 (20 分)

Posted 2021-12-04 kingwzun

题目:

7-8 中序遍历树并判断是否为二叉搜索树 (20 分)
对给定的有N个节点（N>=0）的二叉树，给出中序遍历序列，并判断是否为二叉搜索树。

题目保证二叉树不超过200个节点，节点数值在整型int范围内且各不相同。

输入格式:
第一行是一个非负整数N，表示有N个节点
第二行是一个整数k，是树根的元素值
接下来有N-1行，每行是一个新节点，格式为 r d e 三个整数，
r表示该节点的父节点元素值（保证父节点存在）；d是方向，0表示该节点为父节点的左儿子，1表示右儿子；e是该节点的元素值
输出格式:
首先输出二叉树的中序遍历序列，每个元素占一行。对于空树，不输出任何内容。
然后如果给定的树是二叉搜索树，输出Yes 否则输出No
输入样例:

对于图片中的二叉树：

3
20
20 0 10
20 1 25
结尾无空行

输出样例:

10
20
25
Yes
结尾无空行

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
struct node{
	int var;
	node *L=NULL;
	node *R=NULL;
};
map<int ,node*> mp;// map存当前值的位置，用于快速定位树的位置
int n, root;
int r, d, e;
//r表示该节点的父节点元素值（保证父节点存在）；
//d是方向，0表示该节点为父节点的左儿子，1表示右儿子；
//e是该节点的元素值
void insert(node *&p, int r, int d, int e){
	if(p == NULL){
		p = new node();
		p->var = e;
		mp[e] = p; 
		return ;
	}
	if(p->var == r){
		if(d == 0){
			insert(p->L, r, d, e);
		}
		else{
			insert(p->R, r, d, e);
		} 
	}
}

bool flag = true;
void dfs(node *p){
	if(p->L != NULL){
		if(p->L->var > p->var)flag = false;
		dfs(p->L);
	}
	
	cout << p->var << endl;
	
	if(p->R != NULL){
		if(p->R->var < p->var)flag = false;
		dfs(p->R);
	}
}

int main(){

	node *p = NULL;
	cin >> n;
	if(n == 0){
		cout << "Yes"<<endl;
		return 0;
	}

	cin >> root;
	insert(p,0,0,root);
	for(int i = 1; i < n; i ++){
		cin >> r >> d >> e;
		insert(mp[r], r, d, e);
	}

	dfs(p);

	if(flag == true)cout << "Yes\\n";
	else cout << "No\\n";
	return 0;
}