105 construct tree with preorder and inorder

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;
  }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容