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

给你 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] 之间。

思路：

遍历两个二叉树，然后排序。时间复杂度$O(lo{g}_{2}N)$,或者得到两个有序数组然后归并就是$O(N)$。

class Solution:
    def getAllElements(self, root1: TreeNode, root2: TreeNode) -> List[int]:
        p=root1
        q=root2
        res=[]
        s1=[]
        s2=[]
        while s1 or p or s2 or q:
            while(p):
                s1.append(p)
                p=p.left
            while(q):
                s2.append(q)
                q=q.left
            if s1:
                p=s1.pop()
                res.append(p.val)
                p=p.right
            if s2:
                q=s2.pop()
                res.append(q.val)
                q=q.right
        res.sort()
        return res