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

二叉树的最大路径和

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

相关推荐

Rena1ssance_:对的,要是面评没太烂,勤更新简历等捞就行了,腾讯可以无限复活
点赞 评论 收藏
分享
05-09 13:22
门头沟学院 Java
点赞 评论 收藏
分享
评论
4
1
分享

创作者周榜

更多
牛客网
牛客企业服务