建二叉树
代码中的dfs(node* root) 函数我们只需要改变输出的位置就可以实现三种遍历序列的打印
紧接上篇,代码内容几乎相同,给出先序 + 中序建立二叉树
下面展示 代码
。
#include "bits/stdc++.h"
using namespace std;
const int N = 100;
int pre[N];
int rev[N];
struct node{
node* lchild;
node* rchild;
int val;
};
int p;
node* build(node* root, int s, int e){
if(s > e) return root;
root = new node;
root->lchild = root->rchild = NULL;
root->val = pre[p];
int t;
for(int i = s; i <= e; i++){
if(rev[i] == pre[p]){
t = i;
break;
}
}
p++; // 和后序不同在这里我们从0开始, 一直++
if(s == e) return root;
root->lchild = build(root->lchild, s, t - 1);
root->rchild = build(root->rchild, t + 1, e);
return root;
}
void dfs(node *root){
if(root == NULL) return;
cout<<root->val<<endl; // 先序
if(root->lchild) dfs(root->lchild);
// cout<<root->val<<endl; //中序
if(root->rchild) dfs(root->rchild);
//cout<<root->val<<endl; //后序
}
int main(){
int n;
cin>>n;
for(int i = 0; i < n; i++) cin>>pre[i];
for(int i = 0; i < n; i++) cin>>rev[i];
node* root = NULL;
p = 0;
root = build(root, 0, n - 1);
dfs(root);
}
/*
5
1 2 3 4 5
2 3 1 4 5
*/