文章目錄

• 題目描述
• 思路 & 代碼
• • • 更新版

題目描述

• 好家伙我可太激動了，第一次光速A困難題。
• 應該是因為寫過類似的二叉樹的直徑，不過這道倒是屬于簡單題。。
• 面試常考題，刷codetop看到了就寫寫
  

思路 & 代碼

• 遞歸過程是自底向上的
• 遞歸過程，求出當前最大值 ans，維護 max 用于返回答案
• 遞歸過程，求出當前還可繼續(xù)走的最大路徑 forPa，用于父結(jié)點繼續(xù)使用
• 時間復雜度 O(n)，相當于每個結(jié)點走了一趟

/*** 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 {// 在遞歸過程中，維護一個需要返回的 max 值int max = Integer.MIN_VALUE;public int maxPathSum(TreeNode root) {forMax(root);return max;}int forMax(TreeNode root){if(root == null){return Integer.MIN_VALUE;}// 先往左右子結(jié)點走int left = forMax(root.left);int right = forMax(root.right);// ans：用于維護 max，可能已經(jīng)不能往上走了（已經(jīng)走了左右子樹）int ans = root.val;// 不一定取，因為可能是負數(shù)ans += left > 0 ? left : 0;ans += right > 0 ? right : 0;max = Math.max(ans, max);// forPa：用于給上層結(jié)點，一定不是同時走了左右子樹的路徑int forPa = root.val;// 最多只能走一邊：走左 or 右 or 不走int add = left > right ? left : right;if(add > 0){forPa += add;}// 返回往上走的可行路徑最大值return forPa;} }

更新版

• 照例感嘆之前寫的代碼怎么這么冗余。。

class Solution {int max = Integer.MIN_VALUE;public int maxPathSum(TreeNode root) {maxPath(root);return max;}public int maxPath(TreeNode root) {if(root == null) {return 0;}int left = Math.max(0, maxPath(root.left));int right = Math.max(0, maxPath(root.right));max = Math.max(root.val + left + right, max);return Math.max(left + root.val, right + root.val);} }

總結(jié)

以上是生活随笔為你收集整理的【LeetCode笔记】124. 二叉树中的最大路径和（Java、二叉树、DFS）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。