左神提升5:树型DP 问题

这篇博客介绍了树型动态规划(DP)的基本套路,包括从局部到整体的处理方式和信息传递。以LeetCode的543题《二叉树的直径》为例,阐述了解题思路:考虑节点是否在直径路径上,计算经过和不经过自身的最大直径。通过递归实现,结合左右子树信息得到最大直径。同时,定义了辅助类Info存储高度和直径信息。代码中展示了如何通过树型DP解决二叉树问题。
摘要由CSDN通过智能技术生成

树型DP

关于树型DP的套路

1)从小树到大树,得到答案的可能性,
想出题目的所有可能性,左树右树的所有情况,
2)列出所有的信息 =》 设计Info类
3)写递归

写递归的过程:

  1. 得到info
  2. process使用info进行信息传递
  3. baseCase使用null
  4. 综合leftInfo、rightInfo
  5. 加工左右信息(注意null的时候,什么时候存在)

利用递归的思想来实现dp,dp有空间换时间
在这里插入图片描述

判断是否BST:
先列条件,和套路一样
在这里插入图片描述

543. 二叉树的直径

https://leetcode-cn.com/problems/diameter-of-binary-tree/
点击

解题思路

思路:
分两种情况,经过自己的和没经过自己的。
注意本题的经典套路

1)经过自己:左边的高度和右边的高度,再把自己也加上就是当前的最大直径
2)不经过自己:左边的最大值和右边的最大值加在一起
=》 综合以上来得到最大的值就是本题的答案。

代码

/**
 * 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 {
    public int diameterOfBinaryTree(TreeNode root) {
        return process(root).distance-1;
    }

    public Info process(TreeNode cur) {
        if(cur == null) {
            return null;
        }
        Info leftInfo = process(cur.left);
        Info rightInfo = process(cur.right);

        int height = Math.max(leftInfo == null ?0:leftInfo.height,
                        rightInfo == null ?0:rightInfo.height);

        // 拿到左右两边的最大直径长度(与自己无关的)
        int lm = leftInfo == null ? 0 : leftInfo.distance;
        int rm = rightInfo == null ? 0 : rightInfo.distance;
        int distance = Math.max(lm, rm);
        // 拿到和自己有关的最大直径长度(与自己有关)
        distance = Math.max(((leftInfo == null ?0:leftInfo.height)+
                            (rightInfo == null ?0:rightInfo.height)+1)
                            ,distance);

        return new Info(distance, height+1);
    }



    class Info {
        public int distance;
        public int height;

        Info (int a, int b) {
            distance = a;
            height = b;
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

willorn

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值