题解 | 二叉树的最大路径和

二叉树的最大路径和

http://www.nowcoder.com/practice/da785ea0f64b442488c125b441a4ba4a

分析

读题:

  • 题目难点在于正确理解题意

  • 一棵二叉树

  • 注意题目对路径的定义:开始和结束结点可以是任意的结点。

  • 路径要求唯一,不能重复

  • 任意给出一棵二叉树的两个结点,路径指的是:分别从这两个结点向上走,找到 最近的公共祖先 结点而形成的路径。只有这样的定义下,路径才是唯一确定的。

考虑如下路径:
图片说明

下图,带X的部分路径表示不符合条件

图片说明

解法一:DFS(递归)

思路步骤:

  • 声明int ans = Integer.MIN_VALUE存储答案

  • DFS计算每一个结点的贡献值

  • 由于路径唯一,在回溯时,只需要取某个父节点的俩孩子结点中的最大者。

  • 对于某个节点,它关心自己走入一个子树,能从中捞取的最大收益,先不用管具体怎么走。

  • 定义dfs函数:返回当前子树能向父节点“提供”的最大路径和。即,一条从父节点延伸下来的路径,能在当前子树中获得的最大收益。分为三种情况:

    • 路径停在当前子树的根节点,在这个子树中贡献:root.val
    • 走入左子树,在这个子树中的最大贡献:root.val + dfs(root.left)
    • 走入右子树,在这个子树中的最大贡献:root.val + dfs(root.right)
  • 对应了前面所讲的三种选择,最大贡献取三者最大:
    即:root.val+max(dfs(root.left),dfs(root.right))

Java参考代码:

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @return int整型
     */
    int ans = Integer.MIN_VALUE;
    public int maxPathSum (TreeNode root) {
        // write code here

        if(root==null){
            return 0;
        }
        dfs(root);
        return ans;
    }
    //DFS函数
    public int dfs(TreeNode root){
        if(root==null){
            return 0;
        }
        //计算左孩子贡献
        int leftMax = Math.max(dfs(root.left),0);
        //计算右孩子贡献
        int rightMax = Math.max(dfs(root.right),0);
        //更新答案
        ans = Math.max(ans,root.val+leftMax+rightMax);
      

剩余60%内容,订阅专栏后可继续查看/也可单篇购买

小白专属-牛客题解 文章被收录于专栏

专注于牛客平台编程题题解,文字+图解。内容很细,小白友好系列

全部评论
题目不是也要求空间复杂度是O(1)吗?
点赞 回复 分享
发布于 2022-02-09 09:19

相关推荐

听说改名字就能收到offer哈:Radis写错了兄弟
点赞 评论 收藏
分享
如题,字节跳动怎么才能看到自己的面评,找hr说看不到
SoulStar:自己应该看不到,这个是字节比较保密的信息,之前有mentor加我,说他能看到,但是不能给我说,给我说了他可能就要被辞退了
点赞 评论 收藏
分享
3 1 评论
分享
牛客网
牛客企业服务