[路飞]_js算法:从前序与中序遍历序列构造二叉树

106 阅读1分钟

从前序与中序遍历序列构造二叉树

问题描述: 给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。(by leetcode 105

  示例 1:

输入 : preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

思路: 采用递归的方式,前序遍历第一个参数就是根节点,并在中序中找到,每次从上往下建立节点。每次递归要改变当棵树的前序及中序参数。

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {number[]} preorder
 * @param {number[]} inorder
 * @return {TreeNode}
 */
var buildTree = function(preorder, inorder) {
if(!preorder.length) return null;
let pos=0;
while(preorder[0]!==inorder[pos])pos++;
const node=new TreeNode(preorder[0]);
node.left=buildTree(preorder.slice(1,pos+1),inorder.slice(0,pos))
node.right=buildTree(preorder.slice(pos+1),inorder.slice(pos+1))
return node
};