2022-05-01每日一题

最新推荐文章于 2024-09-13 19:08:00 发布

卍暴力出奇迹卍

最新推荐文章于 2024-09-13 19:08:00 发布

阅读量113

点赞数

分类专栏： leetcode每日一题文章标签：二叉搜索树中序遍历归并排序层序遍历有序数组

本文链接：https://blog.csdn.net/qq_43610675/article/details/124534164

版权

leetcode每日一题专栏收录该内容

43 篇文章 0 订阅

订阅专栏

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] 范围内
-10⁵ <= Node.val <= 10⁵

思路：我是直接进行两次层序遍历（层序遍历主要是利用队列），层序遍历期间把数字都加入一个List中，然后利用Collections的sort方法直接排序，就能得到结果，但这种方式不是最优的
官方解答是利用了二叉搜索树和中序遍历的关系，通过中序遍历得到两个有序数组，之后再用归并的思想完成排序，具体可以参考这个
我的代码

/**
 * 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> list = new ArrayList<>();
        Queue<TreeNode> queue =  new  LinkedList<>();
        if(root1!=null){
            queue.add(root1);
        }
        while(!queue.isEmpty()){
            TreeNode t = queue.poll();
            list.add(t.val);
            if(t.left!=null){
                queue.add(t.left);
            }
            if(t.right!=null){
                queue.add(t.right);
            }
        }
        if(root2!=null){
            queue.add(root2);
        }
        while(!queue.isEmpty()){
            TreeNode t = queue.poll();
            list.add(t.val);
            if(t.left!=null){
                queue.add(t.left);
            }
            if(t.right!=null){
                queue.add(t.right);
            }
        }
        Collections.sort(list);
        return  list;
    }
}

官方解答代码

/**
 * 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> nums1 = new ArrayList<Integer>();
        List<Integer> nums2 = new ArrayList<Integer>();
        inorder(root1, nums1);
        inorder(root2, nums2);

        List<Integer> merged = new ArrayList<Integer>();
        int p1 = 0, p2 = 0;
        while (true) {
            if (p1 == nums1.size()) {
                merged.addAll(nums2.subList(p2, nums2.size()));
                break;
            }
            if (p2 == nums2.size()) {
                merged.addAll(nums1.subList(p1, nums1.size()));
                break;
            }
            if (nums1.get(p1) < nums2.get(p2)) {
                merged.add(nums1.get(p1++));
            } else {
                merged.add(nums2.get(p2++));
            }
        }
        return merged;
    }

    public void inorder(TreeNode node, List<Integer> res) {
        if (node != null) {
            inorder(node.left, res);
            res.add(node.val);
            inorder(node.right, res);
        }
    }
}