写一个函数,输入一个二叉树,树中每个节点存放了一个整数值,函数返回这棵二叉树中相差最大的两个节点间的差值绝对值。

阿里2015笔试附加题1:写一个函数,输入一个二叉树,树中每个节点存放了一个整数值,函数返回这棵二叉树中相差最大的两个节点间的差值绝对值。

【思路】

采用递归实现。深度遍历一遍,用 dp 数组记录遍历过程中最大值(dp[1])和最小值(dp[1]),最后,dp[1] - dp[0]就是所求的:

public class Solution {
    public void dfs(TreeNode root, int[] dp) {
        if (root == null) return;
        dp[0] = Math.min(dp[0], root.val);
        dp[1] = Math.max(dp[1], root.val);
        dfs(root.left, dp);
        dfs(root.right, dp);
    }
    public static void main(String[] args) {
        int[] dp = new int[2];
        dp[0] = Integer.MAX_VALUE;  //存储节点中最小值
        dp[1] = Integer.MIN_VALUE;  //存储节点中最大值
        TreeNode root = new TreeNode(-1);
        TreeNode left = new TreeNode(5);
        TreeNode right = new TreeNode(12);
        left.left = new TreeNode(0);
        left.right = new TreeNode(-4);
        root.left = left;
        right.left = new TreeNode(3);
        right.right = new TreeNode(4);
        root.right = right;
        new Solution().dfs(root, dp);
        System.out.println(dp[1] - dp[0]);
    }
}
结果:

16



你可能感兴趣的:(写一个函数,输入一个二叉树,树中每个节点存放了一个整数值,函数返回这棵二叉树中相差最大的两个节点间的差值绝对值。)