美文网首页
105 construct tree with preorder

105 construct tree with preorder

作者: Fei_JOB | 来源:发表于2017-10-24 03:30 被阅读0次
class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        if(preorder == null || inorder == null || preorder.length == 0 || preorder.length != inorder.length) return null;
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        for(int i = 0; i < inorder.length; i++){
            map.put(inorder[i], i);
        }
        return helper(preorder, 0, preorder.length-1, inorder, 0, inorder.length-1, map);
    }  
    
  private TreeNode helper(int[] preorder, int pS, int pE, int[] inorder, int iS, int iE, Map<Integer, Integer> map){
      if(pS > pE) return null;
      TreeNode node = new TreeNode(preorder[pS]);
      if(pS == pE) return node;
      int idx = map.get(node.val);
      int  leftSize = idx - iS;
      TreeNode left = helper(preorder, pS + 1, pS+leftSize, inorder, iS, idx-1, map);
      TreeNode right = helper(preorder, pS + leftSize + 1, pE, inorder, idx + 1, iE, map);
      node.left = left;
      node.right = right;
      return node;
  }
}

相关文章

网友评论

      本文标题:105 construct tree with preorder

      本文链接:https://www.haomeiwen.com/subject/wjozuxtx.html