leetcode 124 二叉树中的最大路径和

5 阅读1分钟

124. 二叉树中的最大路径和 - 力扣(LeetCode)

可以用递归解决。返回值是max(当前节点值加max(max(左链),max(右链)),0) (注意这里和 0 取最大值了)如果两边子树都是负数就取0)

递归当中用ans更新最大值,更新时跟(当前节点值+左最值+右最值)的和来比较。

代码Java

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    private int ans = Integer.MIN_VALUE;
    public int maxPathSum(TreeNode root) {
        dfs(root);
        return ans;
    }

    public int dfs(TreeNode root) {
        if(root == null)
        return 0;
        int v = root.val;
        int l_val = dfs(root.left);
        int r_val = dfs(root.right);

        ans = Math.max(l_val+r_val+v, ans);
        return Math.max(Math.max(l_val, r_val) + v, 0); // 当前子树最大链和(注意这里和 0 取最大值了)如果两边子树都是负数就取0
    }
}