leetcode 1305.两颗二叉搜索树中的所有元素(js)

题目

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

示例 1：

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

示例 2：

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

提示：

每棵树的节点数在 [0, 5000] 范围内
-105 <= Node.val <= 105

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/all-elements-in-two-binary-search-trees

解题

var getAllElements = function(root1, root2) {
     let nums1=[],nums2=[];
     //二叉搜索树中序遍历模板
     const dfs = (root,res) => {
        if (root === null) {
            return;
        }
        dfs(root.left,res);
        res.push(root.val);
        dfs(root.right,res);
    }
    //获取两个树的中序遍历,中序遍历完的所有整数就是升序排列，这个是二叉搜索树的性质
    dfs(root1,nums1);
    dfs(root2,nums2);
    
    let p1=0,p2=0;
    let ans = [];
    //合并中序遍历
    while(1){
        //当两个中序遍历短的遍历完之后，直接把另外一个长的遍历完，结束循环
        if (p1 === nums1.length) {
            for (let i = p2; i < nums2.length; i++) {
                ans.push(nums2[i]);
            }
            break;
        }
        if (p2 === nums2.length) {
            for (let i = p1; i < nums1.length;i++) {
                ans.push(nums1[i]);
            }
            break;
        }
        //比较值大小，小的值放在前面
        if(nums1[p1]<nums2[p2]){
            ans.push(nums1[p1++]);
        }else{
            ans.push(nums2[p2++]);
        }
    }
    return ans;
};