【力扣每日一题】1305. 两棵二叉搜索树中的所有元素

题目描述

给你 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

---

中序遍历这咱会，但如何将两个数组合并成一个有序数组呢？我当时脑子一热，想直接把一个拼在另一个后面，然后整个重新排序，这样的结果当然是超时。

看了官方的解法后顿悟，既然这个二叉搜索树中序遍历后本来就是升序数组，那么两个升序数组合并成一个升序数组，只要遍历，对首元素不断比较，把小的加入答案数组，不就可以了吗？这本质上是双指针的运用。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<Integer> getAllElements(TreeNode root1, TreeNode root2) {
        List<Integer> ls1 = new ArrayList();
        List<Integer> ls2 = new ArrayList();
        dfs(root1,ls1);
        dfs(root2,ls2);

        List<Integer> merge = new ArrayList();
        int p1=0,p2=0;
        while(true){
            if(p1==ls1.size()){
                merge.addAll(ls2.subList(p2,ls2.size()));
                break;
            }
            if(p2==ls2.size()){
                merge.addAll(ls1.subList(p1,ls1.size()));
                break;
            }
            if(ls1.get(p1)<ls2.get(p2)){
                merge.add(ls1.get(p1++));
            }else{
                merge.add(ls2.get(p2++));
            }
        }
        return merge;
    }
    public void dfs(TreeNode root,List<Integer> ls){
        if (root!=null){
            dfs(root.left,ls);
            ls.add(root.val);
            dfs(root.right,ls);
        }
    }

    
}

---

排序问题不要想得那么死板，不能脑子里只有那几大排序，很多时候还是要利用题目的性质，做出最快的排序选择。