阿里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;
righ