网易 二叉树

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

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
    public TreeNode(int val) {
        this.val = val;
    }
}*/
public class Tree {
     TreeNode maxNode = new TreeNode(Integer.MIN_VALUE);
    TreeNode minNode = new TreeNode(Integer.MAX_VALUE);
    int steps;

    public int getDis(TreeNode root) {
        getMaxMinNode(root);
        TreeNode father = lca(root, minNode, maxNode);
        steps = 0;
        getSteps(father, maxNode, 0);
        int maxSteps = steps;
        steps = 0;
        getSteps(father, minNode, 0);
        int minSteps = steps;
        return maxSteps + minSteps;
    }

    public void getMaxMinNode(TreeNode root) {
        if (root == null) {
            return;
        }
        if (root.left == null && root.right == null) {
            if (root.val > maxNode.val) {
                maxNode = root;
            }
            if (root.val < minNode.val) {
                minNode = root;
            }
        }
        getMaxMinNode(root.left);
        getMaxMinNode(root.right);
    }

    public void getSteps(TreeNode root, TreeNode node, int level) {
        if (root == null) {
            return;
        }
        if (root == node) {
            steps = level;
            return;
        }
        getSteps(root.left, node, level + 1);
        getSteps(root.right, node, level + 1);
    }

    public TreeNode lca(TreeNode root, TreeNode node1, TreeNode node2) {
        if (root == null) {
            return root;
        }
        boolean exist1 = hasNode(root.left, node1);
        boolean exist2 = hasNode(root.left, node2);
        if (exist1 && exist2) {
            return lca(root.left, node1, node2);
        }
        exist1 = hasNode(root.right, node1);
        exist2 = hasNode(root.right, node2);
        if (exist1 && exist2) {
            return lca(root.right, node1, node2);
        }
        return root;
    }

    public boolean hasNode(TreeNode root, TreeNode node) {
        if (root == null || node == null) {
            return false;
        }
        if (root == node) {
            return true;
        }
        return hasNode(root.left, node) || hasNode(root.right, node);
    }

}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值