建二叉树

代码中的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
*/