LeetCode面试298,二叉树最长连续序列(Python)

发布于:2024-05-07 ⋅ 阅读:(23) ⋅ 点赞:(0)

开始想着dfs,两种情况

1.以root为根

2.不以root为根

但是这样需要两个dfs分别进行,那么时间复杂度就上去了。

class Solution:
    def longestConsecutive(self, root: Optional[TreeNode]) -> int:
        def dfs(root):
            # 以root为根节点,可以延续几个点
            ans = 1
            if root.left and root.left.val - root.val == 1:
                ans = max(ans, 1+dfs(root.left))
            if root.right and root.right.val - root.val == 1:
                ans = max(ans, 1+dfs(root.right))
            return ans
        if not root: return 0
        ans = dfs(root)
        ans = max(ans, self.longestConsecutive(root.left), self.longestConsecutive(root.right))
        return ans

题解中提出一种思路可以同时进行dfs,只用把所有节点遍历一遍。

从上到下遍历,建立一个dfs(u, v, length),u为v的父节点,v为u的子节点,length以父节点为最后一个节点的序列长度(初始长度为1)

如果子节点刚好比父节点大1,那么length + 1,反之,length = 1

再继续遍历v的子节点

class Solution:
    def __init__(self):
        self.m = 1

    def longestConsecutive(self, root: Optional[TreeNode]) -> int:
        self.dfs(root, root, 1)
        return self.m

    def dfs(self, u, v, length):
        # u为父节点,v为子节点,length为初始长度
        if not v: return 
        if v.val - u.val == 1:
            length += 1
            self.m = max(self.m, length)
        else:
            length = 1
        self.dfs(v, v.left, length)
        self.dfs(v, v.right, length)

举一反三,还是按照之前的想法,不过只需要改变一点点,只有一种情况:

以root为根的最长序列

如果root.left.val - root.val==1,那么(以root为根的最长序列)可能就等于(以root.left为根的最长序列+1),如果root.right.val - root.val==1,那么(以root为根的最长序列)可能就等于(以root.right为根的最长序列+1),如果都不满足,那么返回1。

怎么有点像列表维护最长序列。其实就可以将链表想象成多方向的列表。

class Solution:
    def __init__(self):
        self.m = 1

    def longestConsecutive(self, root: Optional[TreeNode]) -> int:
        self.dfs(root)
        return self.m

    def dfs(self, root):
        if not root:return
        l, r = self.dfs(root.left), self.dfs(root.right)
        if root.left and root.left.val - root.val == 1:
            l += 1
            self.m = max(self.m, l)
        else:
            l = 0
        if root.right and root.right.val - root.val == 1:
            r += 1
            self.m = max(self.m, r)
        else:
            r = 0
        return max(1, l, r)