如何通过数组构造二叉树?

对于一个二叉树,我们一般有三种不同的遍历方式,分为先序遍历,中序遍历,后序遍历。

先序遍历:根节点 -> 左子树 -> 右子树

中序遍历:左子树 -> 根节点 -> 右子树

后序遍历:左子树 -> 右子树 -> 根节点

根据数组(数组中的值各不相同)构造一个二叉树,我们一般有两种方式,其一是从中序与后序遍历序列构造二叉树(106. 从中序与后序遍历序列构造二叉树),二是从前序与中序遍历序列构造二叉树(105. 从前序与中序遍历序列构造二叉树)。本文以中序与后序为例,我们看如下二叉树和数组。

  1.  构造一棵二叉树,我们首先要确定的就是节点的值,中序为左根右的方式,而后序为左右根的方式,也就是说,后序数组的最后一个元素即为根元素。
  2. 根据这样的根元素,我们又可以将中序分为左子树区间,右子树区间,对于后序数组串联在一起的左右区间,我们可以依据中序数组的左子树区间中数的个数将后序数组切分为左区间和右区间。
  3. 对于切分出来的左区间,后序数组的最后一个元素仍为该左子树的根节点,根据该根节点切分中序数组,得到左左区间,左右区间。
    在切分过程中,我们需要接住叶子节点,比如上例,在进行第二次切分时候他的左区间就仅剩下一个,这时候我们可以直接返回,不再对该区间进行操作。
  4. 重复以上步骤,利用递归的思想,我们最终可以构造出一颗二叉树。

解题代码:

#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
using namespace std;


struct TreeNode {
	int val;
	TreeNode *left;
	TreeNode *right;
}; 

TreeNode* traversal(vector<int>& inorder, vector<int>& postorder){
	if(postorder.size() == 0) return NULL;
	TreeNode* root = new TreeNode;	// 初始化根节点,并且分配一个内存空间,否则接下来对它进行操作,可能会导致未定义的行为 
	root->val = postorder[postorder.size() - 1];
	if(postorder.size() == 1) return root; // 遍历到了叶子节点,直接返回

	// 找到根节点在中序数组的下标,用于切分左右区间 
	vector<int>::iterator it = find(inorder.begin(), inorder.end(), postorder[postorder.size() - 1]);
    int index = distance(inorder.begin(), it);
    
    // 切分左右区间
    // 中序 
    vector<int> left_inorder(inorder.begin(), inorder.begin() + index);
    vector<int> right_inorder(inorder.begin() + index + 1, inorder.end());
    // 后序 
    vector<int> left_postorder(postorder.begin(), postorder.begin() + index);
    vector<int> right_postorder(postorder.begin() + index, postorder.end() - 1);
    
    // 设置左节点和右节点 
	root->left = traversal(left_inorder, left_postorder);
	root->right = traversal(right_inorder, right_postorder);
	
	return root; 
}

int main(){
	int arri[] = {9,3,15,20,7};
    vector<int> inorder(arri, arri + sizeof(arri) / sizeof(arri[0]));
    int arrp[] = {9,15,7,20,3};
    vector<int> postorder(arrp, arrp + sizeof(arrp) / sizeof(arrp[0]));
	traversal(inorder, postorder);
	return 0;
}

 

 

 

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

zzr-rr

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值