网易笔试题:二叉树

有一棵二叉树,树上每个点标有权值,权值各不相同,请设计一个算法算出权值最大的叶节点到权值最小的叶节点的距离。二叉树每条边的距离为1,一个节点经过多少条边到达另一个节点为这两个节点之间的距离。
给定二叉树的根节点root,请返回所求距离。

思路:分两步
1.找到权值最大的位置和权值最小的位置;2.将它们到最近公共父节点的距离相加就能得到结果。

public static int getDis(TreeNode root) {
        int maxWeight = Integer.MIN_VALUE; //最大权值
        int minWeight = Integer.MAX_VALUE; //最小权值
        // 建立权重、结点编号的map
        HashMap map1 = new HashMap();
        LinkedList queue1 = new LinkedList();
        // 记录子、父节点的权重
        HashMap map2 = new HashMap();
        queue1.addFirst(root);
        int count = 0;
        map1.put(root.val, count++);
        while (!queue1.isEmpty()) {
            TreeNode temp = queue1.poll();
            if (temp.left != null) {
                queue1.addLast(temp.left);
                map1.put(temp.left.val, count++);
                map2.put(temp.left.val, temp.val);
            }
            if (temp.right != null) {
                queue1.addLast(temp.right);
                map1.put(temp.right.val, count++);
                map2.put(temp.right.val, temp.val);
            }
            if (temp.left == null && temp.right == null) { //叶子结点
                if (temp.val > maxWeight) {
                    maxWeight = temp.val;
                }
                if (temp.val < minWeight) {
                    minWeight = temp.val;
                }
            }
        }
        // 回溯的终止条件是回溯到同一个节点 此时 maxWeight = minWeight 
        int len1 = 0;
        int len2 = 0;
        while (maxWeight != minWeight) {
            if (map1.get(maxWeight) > map1.get(minWeight)) { 
                maxWeight = map2.get(maxWeight);
                len1++;
            } else if (map1.get(maxWeight) < map1.get(minWeight)) {
                minWeight = map2.get(minWeight);
                len2++;
            } else { //此时的父节点是公共父节点
                maxWeight = map2.get(maxWeight);
                minWeight = map2.get(minWeight);
                len1++;
                len2++;
            }
        }
        return len1 + len2;

    }

你可能感兴趣的:(经典算法面试题,二叉树,网易)