LeetCode124.二叉树中最大路径和

第一次只花了20分钟左右就完全靠自己把一道hard题做出来了。我这个方法还是非常简单非常容易理解的,虽然时间复杂度达到了O(n2)。以下是我的代码:

java 复制代码
class Solution {
    int max;
    public int maxPathSum(TreeNode root) {
        max = Integer.MIN_VALUE;
        return dfs2(root);
    }
    public int dfs2(TreeNode root){
        if(root == null)return Integer.MIN_VALUE;
        max = Math.max(dfs(root), dfs(root.left)+dfs(root.right)+root.val);
        return Math.max(max, Math.max(dfs2(root.left), dfs2(root.right)));
    }
    public int dfs(TreeNode root){
        if(root == null){
            return 0;
        }
        return root.val + Math.max(0,Math.max(dfs(root.left), dfs(root.right)));
    }
}

我用了两个深度优先遍历,首先第一个dfs(TreeNode root)方法是找出以root的为根节点的单向最大和的路径。

这个非常好求,root为根的单向最大和路径就是(左子树为根的单向最大路径和右子树为根的单向最大路径中的最大的+root.val),当然如果两个子节点的最大和路径都是负数,那么root它自己单独就是最大的。

第二个深度优先遍历dfs2(TreeNode root)方法是遍历整颗树的节点,找出分别以每个节点的最大和路径 (不只是单向的,还可以通过根节点把左右两边连起来)的最大值,返回最大值即可。

这个也非常好求,要么是以root为根的单向的最大值dfs(root) ,要么是把根节点和左边最大的和右边最大的连起来dfs(root.left)+dfs(root.right)+root.val。为什么不考虑只连左边或者只连右边呢?因为第一种情况dfs(root)就是考虑了只连左或者只连右后者单独一个节点。然后进行递归找出最大值即可return Math.max(max, Math.max(dfs2(root.left), dfs2(root.right)))。

看看官方题解做法吧

看完题解我觉得我像个傻逼,一次深度遍历dfs就好了,只需要设置一个全局变量max,然后每遍历到一个节点就更新一下max就好了,所以把我的代码稍微改一下就是题解代码:

java 复制代码
class Solution {
    int max;
    public int maxPathSum(TreeNode root) {
        max = Integer.MIN_VALUE;
        dfs(root);
        return max;
    }
    public int dfs(TreeNode root){
        if(root == null){
            return 0;
        }
        int lefeCon = Math.max(0,dfs(root.left));
        int rightCon = Math.max(0, dfs(root.right));
        max = Math.max(max, root.val+lefeCon+rightCon);
        
        return root.val + Math.max(0,Math.max(lefeCon, rightCon));
    }
}

我这个dfs(TreeNode root)还是返回根节点连上左边或者连上右边的最大值(不是同时连左边和右边),然后用max是左右都连起来,max不停更新,最后返回max即可。

相关推荐
FinelyYang11 分钟前
考试系统简答题 AI 阅卷:覆盖度驱动、异步评分与人机协同落地实践
java·大数据·人工智能
hhhjhl17 分钟前
Yup 跨端三方库仓颉适配全流程
java·开发语言
用户31268748772036 分钟前
Java 对象内存布局:压缩指针、对齐填充与 JOL 实测
java
东华果汁哥1 小时前
【Spring AI 实战】从零开始调用大模型接口,构建 Java AI 应用
java·人工智能·spring
深圳市益普科技有限公司1 小时前
半导体良率管理YMS:良率与设备参数的相关性,先拉哪几组数据
人工智能·python·算法
Song_RsHt1 小时前
Redis 与微服务,什么场景下无可替代
java·redis·后端·学习·微服务
学长毕设2 小时前
基于SpringBoot的建筑材料管理系统-附源码
java·spring boot·后端
摸爬滚打的小李2 小时前
关于 Arthas
java·后端
程序员清风2 小时前
算法题中的边界条件、测试与调试
开发语言·算法·kotlin
砚底藏山河2 小时前
python量化实操:冷热分层存储
java·开发语言·python·金融