【二叉树-中等】1372. 二叉树中的最长交错路径

最新推荐文章于 2025-12-18 17:00:31 发布

原创最新推荐文章于 2025-12-18 17:00:31 发布 · 185 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #leetcode #算法

刷题专栏收录该内容

643 篇文章

订阅专栏

这篇博客介绍了如何找到一棵二叉树中最长的交错路径。交错路径指的是从某个节点开始，先经过左子节点再经过右子节点，或者先经过右子节点再经过左子节点的方式。文中提供了三种不同的解决方案，包括超时的递归方法、优化的先序遍历以及深度优先搜索（DFS）方法。这些方法都在保证正确性的前提下，尽可能地提高了效率。

【题目】

每棵树最多有 50000 个节点。
每个节点的值在 [1, 100] 之间。

【代码】
【方法1】超时！！！虽然结果正确，但是时间复杂度高，结果超时，仍然提供出来，作为参考：
在这里插入图片描述

class Solution:
    def visit(self,root,loc=0):
        if not root:
            return 
        self.path.append(loc)
        self.visit(root.left,-1)
        self.visit(root.right,1)
        if not root.left and not root.right and len(self.path)>=3 and len(self.path)>self.ans:
            cnt=0
            for i in range(2,len(self.path)):
                if self.path[i]!=self.path[i-1]:
                    if cnt==0:
                        cnt+=2
                    else:
                        cnt+=1
                else:
                    cnt=0
                    if len(self.path)-i<self.ans:
                        break
                self.ans=max(self.ans,cnt)
        self.path.pop()
    def longestZigZag(self, root: TreeNode) -> int:
        self.ans=0
        if root.left or root.right:
            self.ans=max(self.ans,1)
        self.path=[]
        self.visit(root)
        return self.ans

【方法2】先序遍历，顺带两个参数在函数头，用来记录到该节点为止时，该节点来自父节点的那个分支，以及到该父节点的交错路径的长度
在这里插入图片描述目前发现这个是最简便的写法，还没有发现跟我一样的

class Solution:
    def visit(self,root,loc=0,cnt=0):
        if not root:
            return         
        self.ans=max(self.ans,cnt)
        self.visit(root.left,-1,(cnt+1) if (loc!=-1) else 1)
        self.visit(root.right,1,(cnt+1) if (loc!=1) else 1)        
    def longestZigZag(self, root: TreeNode) -> int:
        self.ans=0
        self.visit(root)
        return self.ans

【方法3】
在这里插入图片描述

class Solution:       
    def longestZigZag(self, root: TreeNode) -> int:
        ans = 0
        def dfs(root, l, r):
            nonlocal ans
            ans = max(ans, l, r)
            if root.left:
                dfs(root.left, r + 1, 0)
            if root.right:
                dfs(root.right, 0, l + 1)
        dfs(root, 0, 0)
        return ans