新手算法學習之路----二叉樹(二叉樹最大路徑和)

摘抄自:https://segmentfault.com/a/1190000003554858#articleHeader2node

題目:segmentfault

Given a binary tree, find the maximum path sum.spa

The path may start and end at any node in the tree.code

For example: Given the below binary tree,blog

 
1 / \ 2 3 

Return 6.遞歸

思路:首先咱們分析一下對於指定某個節點爲根時,最大的路徑和有多是哪些狀況。第一種是左子樹的路徑加上當前節點,第二種是右子樹的路徑加上當前節點,第三種是左右子樹的路徑加上當前節點(至關於一條橫跨當前節點的路徑),第四種是隻有本身的路徑。乍一看彷佛以此爲條件進行自下而上遞歸就好了,然而這四種狀況只是用來計算以當前節點根的最大路徑,若是當前節點上面還有節點,那它的父節點是不能累加第三種狀況的。因此咱們要計算兩個最大值,一個是當前節點下最大路徑和,另外一個是若是要鏈接父節點時最大的路徑和。咱們用前者更新全局最大量,用後者返回遞歸值就好了。get

Java代碼:io

public class Solution {
    
    private int max = Integer.MIN_VALUE;
    
    public int maxPathSum(TreeNode root) {
        helper(root);
        return max;
    }
    
    public int helper(TreeNode root) {
        if(root == null) return 0;
        int left = helper(root.left);
        int right = helper(root.right);
        //鏈接父節點的最大路徑是1、2、四這三種狀況的最大值
        int currSum = Math.max(Math.max(left + root.val, right + root.val), root.val);
        //當前節點的最大路徑是1、2、3、四這四種狀況的最大值
        int currMax = Math.max(currSum, left + right + root.val);
        //用當前最大來更新全局最大
        max = Math.max(currMax, max);
        return currSum;
    }
}
相關文章
相關標籤/搜索