검지offer 두 갈래 나무 재건

2632 단어 두 갈래 나무
제목 설명
두 갈래 나무의 앞 순서와 중간 순서의 결과를 입력하십시오. 이 두 갈래 나무를 다시 만드십시오.입력한 앞 순서와 중간 순서의 결과에 중복된 숫자가 없다고 가정하십시오.예를 들어 앞 순서 반복 시퀀스 {1,2,4,7,3,5,6,8}와 중간 순서 반복 시퀀스 {4,7,2,1,5,3,8,6}를 입력하면 두 갈래 트리를 재건하고 되돌려줍니다.
  public TreeNode reConstructBinaryTree(int[] pre, int[] in) {
        if (pre == null || in == null) {
            return null;
        }
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0, len = in.length; i < len; i++) {
            map.put(in[i], i);
        }
        return creatTree(pre, 0, pre.length - 1, in, 0, in.length - 1, map);
    }

    public TreeNode creatTree(int[] pre, int pi, int pj, int[] in, int ii, int ij, HashMap<Integer, Integer> map) {
        if (pi > pj) {
            return null;
        }
        TreeNode root = new TreeNode(pre[pi]);
        int index = map.get(pre[pi]);
        root.left = creatTree(pre, pi + 1, pi + index - ii, in, ii, index - 1, map);
        root.right = creatTree(pre, pi + index - ii + 1, pj, in, index + 1, ij, map);
        return root;
    }

좋은 웹페이지 즐겨찾기