【路飞】算法与数据结构-两棵二叉搜索树中的所有元素

不管全世界所有人怎么说，我都认为自己的感受才是正确的。无论别人怎么看，我绝不打乱自己的节奏。喜欢的事自然可以坚持，不喜欢的怎么也长久不了。

LeetCode：原题地址

题目要求

给你 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.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root1
 * @param {TreeNode} root2
 * @return {number[]}
 */
var getAllElements = function(root1, root2) {
    function inOrder(root){
        var stack = [];
        var p = root;
        var res = [];
        while(p || stack.length) {
            if (p) {
                stack.push(p);
                p = p.left;
            } else {
                p = stack.pop();
                res.push(p.val);
                p = p.right;
            }
        }
        return res;
    }
    return inOrder(root1).concat(inOrder(root2)).sort((x, y) => x - y);
};