leetcode刷题250天(64)——1305. 两棵二叉搜索树中的所有元素(dfs)

最新推荐文章于 2024-10-06 13:05:10 发布

ウルトラマン.

最新推荐文章于 2024-10-06 13:05:10 发布

阅读量804

点赞数

分类专栏： leetcode 文章标签：深度优先 leetcode 算法

本文链接：https://blog.csdn.net/qq_35668477/article/details/124522546

版权

leetcode 专栏收录该内容

103 篇文章 1 订阅

订阅专栏

给你 root1 和 root2 这两棵二叉搜索树。请你返回一个列表，
其中包含两棵树中的所有整数并按升序排序

解题思路：
最简单的方法就是先遍历再排序
或者
本身每棵树的中序遍历都是有序的，直接双指针滑动比较，存入新list也可以

class Solution(object):
    def getAllElements(self, root1, root2):
        """
        :type root1: TreeNode
        :type root2: TreeNode
        :rtype: List[int]
        """
        res = []

        def dfs(root):
            if not root:
                return 
            
            dfs(root.left)
            res.append(root.val)
            dfs(root.right)

        dfs(root1)
        dfs(root2)

        res.sort()

        return res