【树-中等】1305. 两棵二叉搜索树中的所有元素_1305 两棵二叉搜索树-优快云博客

本文链接：https://blog.youkuaiyun.com/kz_java/article/details/119686962

本文介绍如何通过递归遍历两个二叉搜索树，将所有整数按升序合并为一个有序列表。方法涉及中序遍历和比较节点值的操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【题目】
给你 root1 和 root2 这两棵二叉搜索树。

请你返回一个列表，其中包含两棵树中的所有整数并按升序排序。.

示例 1：

输入：root1 = [2,1,4], root2 = [1,0,3]
输出：[0,1,1,2,3,4]

示例 2：

输入：root1 = [0,-10,10], root2 = [5,1,7,0,2]
输出：[-10,0,0,1,2,5,7,10]

示例 3：

输入：root1 = [], root2 = [5,1,7,0,2]
输出：[0,1,2,5,7]

示例 4：

输入：root1 = [0,-10,10], root2 = []
输出：[-10,0,10]

示例 5：

输入：root1 = [1,null,8], root2 = [8,1]
输出：[1,1,8,8]

提示：

每棵树最多有 5000 个节点。
每个节点的值在 [-10^5, 10^5] 之间。

【代码】
在这里插入图片描述

# 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 getAllElements(self, root1: TreeNode, root2: TreeNode) -> List[int]:
        def inOrder(root):
            return inOrder(root.left)+[root.val]+inOrder(root.right) if root else []
        r1=inOrder(root1)
        r2=inOrder(root2)
        r1.extend(r2)
        r1.sort()
        return r1

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

# 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 getAllElements(self, root1: TreeNode, root2: TreeNode) -> List[int]:
        def inOrder(root):
            return inOrder(root.left)+[root.val]+inOrder(root.right) if root else []
        r1=inOrder(root1)
        r2=inOrder(root2)
        ans=[]
        while r1 and r2:
            if r1[0]<r2[0]:
                ans.append(r1[0])
                r1.pop(0)
            else:
                ans.append(r2[0])
                r2.pop(0)
        if r1:
            ans.extend(r1)
        else:
            ans.extend(r2)
        return ans

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

# 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 getAllElements(self, root1: TreeNode, root2: TreeNode) -> List[int]:
        def inOrder(root):
            return inOrder(root.left)+[root.val]+inOrder(root.right) if root else []
        r1=inOrder(root1)
        r2=inOrder(root2)
        ans=[]
        l1=len(r1)
        l2=len(r2)
        i=0
        j=0
        while i<l1 and j<l2:
            if r1[i]<r2[j]:
                ans.append(r1[i])
                i+=1
            else:
                ans.append(r2[j])
                j+=1
        if i<l1:
            ans.extend(r1[i:])
        else:
            ans.extend(r2[j:])
        return ans