Leetcode 1305.两棵二叉搜索树中的所有元素（All Elements in Two Binary Search Trees）

Leetcode 1305.两棵二叉搜索树中的所有元素

1 题目描述（Leetcode题目链接）

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

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

提示：

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

2 题解

  中序遍历后再归并排序，这里直接sort得了。二叉搜索树相关知识

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def inorder(self, root):
        seq = []
        if root:
            seq += self.inorder(root.left)
            seq.append(root.val)
            seq += self.inorder(root.right)
        return seq

    def getAllElements(self, root1: TreeNode, root2: TreeNode) -> List[int]:
        return sorted(self.inorder(root1) + self.inorder(root2))