两颗二叉搜索树中的所有元素

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

分析：

二叉搜索树：中序遍历结果集为升序结果

题目要求：树合并 并 返回升序结果集

思路一：

1.中序遍历：分别中序遍历两棵树存储为两个升序结果集

2.合并两个升序的结果集

/**
 * 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> r1 = new ArrayList<>();
        List<Integer> r2 = new ArrayList<>();
        inOrder(root1,r1);
        inOrder(root2,r2);
        List<Integer> ret = new ArrayList<>();
        sort(r1,r2,ret);
        return ret;
    }
    private void inOrder(TreeNode root,List<Integer> list){
        if(root==null)
            return ;
        inOrder(root.left,list);
        list.add(root.val);
        inOrder(root.right,list);
    }
    private void sort(List<Integer> l1,List<Integer> l2,List<Integer> ret){
        int size1 = l1.size();
        int size2 = l2.size();
        int i = 0;
        int j = 0;
        while(i<size1 && j<size2){
            int n = l1.get(i);
            int m = l2.get(j);
            if(n<m){
                ret.add(n);
                i++;
            }else{
                ret.add(m);
                j++;
            }
        }
        while(i<size1){
            ret.add(l1.get(i++));
        }
        while(j<size2){
            ret.add(l2.get(j++));
        }
    }
}

 

刷题多一点，大厂近亿点！