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

1，问题简述

给你 root1 和 root2 这两棵二叉搜索树。

请你返回一个列表，其中包含 两棵树 中的所有整数并按 升序 排序。

2，示例

输入：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]

3，题解思路

迭代每棵树的节点值到集合中，然后对集合进行排序，常规思路图

4，题解程序

import java.util.*;


public class GetAllElementsTest2 {
    public static void main(String[] args) {
        TreeNode t11 = new TreeNode(2);
        TreeNode t12 = new TreeNode(1);
        TreeNode t13 = new TreeNode(4);
        t11.left = t12;
        t11.right = t13;


        TreeNode t21 = new TreeNode(1);
        TreeNode t22 = new TreeNode(0);
        TreeNode t23 = new TreeNode(3);


        t21.left = t22;
        t21.right = t23;
        List<Integer> allElements = getAllElements(t11, t21);
        System.out.println("allElements = " + allElements);


    }


    public static List<Integer> getAllElements(TreeNode root1, TreeNode root2) {
        List<Integer> list = new ArrayList<>();
        if (root1 == null && root2 == null) {
            return list;
        }
        List<Integer> list1 = getNode(root1);
        List<Integer> list2 = getNode(root2);
        list.addAll(list1);
        list.addAll(list2);
        Collections.sort(list);
        return list;


    }


    private static List<Integer> getNode(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        List<Integer> list = new ArrayList<>();
        while (!queue.isEmpty()) {
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                if (node != null) {
                    list.add(node.val);
                    if (node.left != null) {
                        queue.add(node.left);
                    }
                    if (node.right != null) {
                        queue.add(node.right);
                    }
                }


            }
        }
        return list;
    }
}

5，题解程序图片版

6，总结

迭代器会使用了吧，这是最常规的解题思路了，没有基于其他方式进行操作，看过的应该可以知道怎么个意思解决了，这就是我输出文章最直接能帮助到需要的人。