題目
給出一棵二叉樹,尋找一條路徑使其路徑和最大,路徑可以在任一節點中開始和結束(路徑和為兩個節點之間所在路徑上的節點權值之和)
樣例 給出一棵二叉樹:
返回 6分析
這道題關於二叉樹的最大路徑問題,顯然需要用遞迴解決。 由於這道題中所求路徑是可以通過根節點的,從任意節點開始和結束的最大路徑和。 顯然,通過根節點的最大路徑=根節點的值,左子樹的最大路徑+右子樹的最大路徑 所以我們可以遞迴求解左右子樹的最大路徑,最後把他們加到一起就是整個樹的最大路徑。
程式碼
/**
* Definition of TreeNode:
* public class TreeNode {
* public int val;
* public TreeNode left, right;
* public TreeNode(int val) {
* this.val = val;
* this.left = this.right = null;
* }
* }
*/
public class Solution {
/**
* @param root: The root of binary tree.
* @return: An integer.
*/
public int maxPathSum(TreeNode root) {
if(root == null)
return 0;
ArrayList<Integer> res = new ArrayList<>();
res.add(Integer.MIN_VALUE);
helper(root, res);
return res.get(0);
}
private int helper(TreeNode root, ArrayList<Integer> res) {
if(root == null)
return 0;
int left = helper(root.left,res);
int right = helper(root.right,res);
int cur = root.val + (left>0?left:0) + (right>0?right:0);
if(cur>res.get(0))
res.set(0, cur);
return root.val + Math.max(left, Math.max(right, 0));
}
}
複製程式碼