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

题目:

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){
		/*改成else也可以
		因为：是用的map定位 数值为e的结点的父亲，
		只有两种情况，
		     一种是p为空（还没有建立map链接），
		     另一种就是定位成功
		*/
		if(d==0){
				p->L=new node();
	        	p->L->var=e;
		        mp[e]=p->L;
		}
		else {
			p->R=new node();
	        	p->R->var=e;
		        mp[e]=p->R;
		}
	}
}
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;
}

简洁代码：

#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;
}