Leetcode 543. 二叉树的直径

在这里插入图片描述

心路历程:

这道题如果题干中不说这条路径可能不经过root的话,可能这道题就是个中等难度题了。思考二叉树问题:如何在递归遍历中解决问题,如何转化为已有的问题。
这道题的特殊之处在于需要对二叉树进行两次的遍历,第一次遍历计算以node结点为根到其叶子结点的最大路径长度,第二次遍历计算以node为根节点的左右子树最大长度的和+2。

注意的点:

1、求的是路径的长度,注意要在结点的数量上减一
2、第二次遍历时记得+2代表该根节点到左右两个子树的距离和
3、二叉树node遍历加上cache装饰器会更快

解法:先后两次dfs遍历

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:

        @cache
        def dfs1(node):  # node结点为根节点到叶子结点的路径数
            if not node: return -1  # 这里-1是为了将结点数转化为边的数量
            return max(dfs1(node.left), dfs1(node.right)) + 1
        
        maxlen = 0
        @cache
        def dfs2(node): # 遍历二叉树
            nonlocal maxlen
            if not node: return
            leftmax = dfs1(node.left)
            rightmax = dfs1(node.right)
            maxlen = max(maxlen, 2 + leftmax + rightmax)
            dfs2(node.left)
            dfs2(node.right)

        dfs2(root)
        return maxlen
  • 10
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值