【105-Construct Binary Tree from Preorder and Inorder Traversal（通过前序和中序遍历构造二叉树）】
【LeetCode-面试算法经典-Java实现】【所有题目目录索引】
原题
Given preorder and inorder traversal of a tree, construct the binary tree.
Note:
You may assume that duplicates do not exist in the tree.
题目大意
给定一个前序和中序遍历序列，构建一个二叉树
注意：
– 二叉树中元素重复元素
解题思路
前序遍历第一个元素是根结点（k），在中序遍历序列中找值为k的下标idx，idx将中序遍历序列分成左右子树，对前序遍历序列也一样，可进行递归操作
代码实现
树结点类
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}算法实现类一：
public class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
if (preorder == null || preorder.length ==0){
return null;
}
HashMap<Integer, Integer> inorderMap = new HashMap<Integer, Integer>();
for (int i=0;i<inorder.length;i++){
inorderMap.put(inorder[i],i);
}
Deque<TreeNode> stack = new LinkedList<TreeNode>();
TreeNode root = new TreeNode(preorder[0]);
stack.push(root);
for (int i=1;i<preorder.length;i++){
TreeNode top = stack.peek();
int indexTop = inorderMap.get(top.val);
int indexVal = inorderMap.get(preorder[i]);
TreeNode node = new TreeNode(preorder[i]);
if (indexVal<indexTop){
top.left=node;
}
else{
while (indexVal>indexTop){
top = stack.pop();
indexTop = stack.isEmpty()?Integer.MAX_VALUE:inorderMap.get(stack.peek().val);
}
top.right = node;
}
stack.push(node);
}
return root;
}
}算法实现类二：（会超时）
public class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
// 参数校验
if (preorder == null || inorder == null || preorder.length == 0
|| preorder.length != inorder.length) {
return null;
}
return solve(preorder, 0, preorder.length - 1, inorder, 0, inorder.length - 1);
}
/** * 构建二叉树，数据输入的正确性由输入数据自己保证 * * @param preorder 先序遍历的结果 * @param x 先序遍历的开始位置 * @param y 先序遍历的结束位置 * @param inorder 中序遍历的结果 * @param i 中序遍历的开始位置 * @param j 中序遍历的结束位置 * @return 二叉树的根结点 */
public TreeNode solve(int[] preorder, int x, int y, int[] inorder, int i, int j) {
if (x >= 0 && x <= y && i >= 0 && i <= j) {
// 只有一个元素
if (x == y) {
return new TreeNode(preorder[x]);
} else if (x < y) {
// 记录根结点的索引
int idx = i;
while (idx <= j && inorder[idx] != preorder[x]) {
idx++;
}
// 创建根结点
TreeNode root = new TreeNode(inorder[idx]);
// 左子树的结点个数
int leftLength = idx - i;
//
if (leftLength > 0) {
// x + 1, x + leftLength：左子树起始和结束位置
root.left = solve(preorder, x + 1, x + leftLength, inorder, i, idx - 1);
}
// 右子树的结点个数
int rightLength = j - idx;
if (rightLength > 0) {
// x + leftLength + 1, y：右子树起始和结束位置
root.right = solve(preorder, x + leftLength + 1, y, inorder, idx + 1, j);
}
return root;
}
}
return null;
}
}评测结果
点击图片，鼠标不释放，拖动一段位置，释放后在新的窗口中查看完整图片。
