前端必考算法——前序中序还原二叉树之代码演示

上一篇：
前端必考算法——给出后序中序还原二叉树

前序中序还原二叉树之代码演示

代码演示：

var preamble = ["a", "c", "f", "g", "b", "d", "e"]; // 前序遍历数组
var middleOrder = ["f", "c", "g", "a", "d", "b", "e"]; // 中序遍历数组

function Node(value) {
  this.value = value;
  this.left = null;
  this.right = null;
}

function fn(preamble, middleOrder) {
  if (preamble === null || middleOrder === null || preamble.length === 0 || middleOrder.length === 0 || preamble.length !== middleOrder.length) return null;
  var root = new Node(preamble[0]); // 根节点是前序遍历的第0个
  var index = middleOrder.indexOf(root.value); // 找到根节点在中序遍历中的所在位置
  var preambleLeft = preamble.slice(1, index + 1); // 前序遍历的左子树
  var preambleRight = preamble.slice(index + 1, preamble.length); // 前序遍历的右子树
  var middleOrderLeft = middleOrder.slice(0, index); // 中序遍历的左子树
  var middleOrderRight = middleOrder.slice(index + 1, middleOrder.length); // 中序遍历的右子树
  root.left = fn(preambleLeft, middleOrderLeft); // 将左子树的前序遍历和中序遍历传入，可以得到中序遍历左子树的根节点
  root.right = fn(preambleRight, middleOrderRight); // 将右子树的前序遍历和中序遍历传入，可以得到中序遍历右子树的根节点
  return root;
}

var root = fn(preamble, middleOrder);
console.log(root.left);
console.log(root.right);

运行结果：

下一篇：
前端必考算法——后序中序还原二叉树之代码演示