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