JavaScript重建二叉树-CSDN博客

本文链接：https://blog.csdn.net/kyr1e/article/details/88200774

要求

根据前序遍历和中序遍历来重建二叉树.

前序遍历: [1,2,4,7,3,5,6,8]
中序遍历: [4,7,2,1,5,3,8,6]
返回二叉树

递归实现

1.前序可得: 根的下一个元素为左子根, 右子根在减去中序所得左子根所有元素后的第一个元素.
2.中序可得: 左子根在根元素的左边, 右子根在根元素的右边.

即根为1,左子根为2,右子根为3
从前序获取根元素.即左子根为preOrder[preStart + 1], 右子根为preOrder[preStart + (i - inStart) + 1].
preStart为根元素在前序序列的索引,i为根元素在中序序列的索引.inStart为左子树根/右子树根的索引

function main() {
    let pre = [1,2,4,7,3,5,6,8];
    let vin = [4,7,2,1,5,3,8,6];
    let result = rebuild(pre, vin, 0, pre.length - 1, 0, vin.length - 1);
    return result;
}

function rebuild(preOrder, inOrder, preStart, preEnd, inStart, inEnd) {
	if(preStart > preEnd || inStart > inEnd)
   		return null;
	let Tree = new TreeNode(preOrder[preStart]);
	for(let i = inStart; i < inEnd; i++) {
		if(preOrder[preStart] === inOrder[i]){
			// left
			Tree.left = rebuild(preOrder, inOrder, 
								preStart + 1, preStart + (i - inStart),
								inStart, i - 1);
			// right
			Tree.right = rebuild(preOrder, inOrder, 
								 preStart + (i - inStart) + 1, preEnd, 
								 i + 1, inEnd);
		}
	}
	return Tree;
}

function TreeNode(x) {
    this.val = x;
    this.left = null;
    this.right = null;
}