【日常训练】1305. 两棵二叉搜索树中的所有元素-优快云博客

本文链接：https://blog.youkuaiyun.com/Puppet__/article/details/124523262

本文介绍了如何使用Java编程语言解决LeetCode 1305题，通过中序遍历合并两个二叉搜索树得到升序排列的整数列表。详细展示了如何遍历并合并两个有序序列，适用于树形数据结构的基础操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

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

代码

package dayLeetCode;

import java.util.ArrayList;
import java.util.List;

public class dayleetcode1305 {
    /**
     * 二叉搜索树是有序的，所以我们先中序遍历得到一个从小到大排序的序列
     * 然后将两个有序序列进行合并
     *
     * @param root1
     * @param root2
     * @return
     */
    public List<Integer> getAllElements(TreeNode root1, TreeNode root2) {
        return getArrays(inorderSearch(root1, new ArrayList<Integer>()), inorderSearch(root2, new ArrayList<Integer>()));
    }

    private List<Integer> getArrays(List<Integer> inorderSearch1, List<Integer> inorderSearch2) {
        List<Integer> ansList = new ArrayList<>();
        int i = 0, j = 0;
        while (i < inorderSearch1.size() && j < inorderSearch2.size()) {
            if (inorderSearch1.get(i) < inorderSearch2.get(j)) {
                ansList.add(inorderSearch1.get(i));
                i++;
            } else {
                ansList.add(inorderSearch2.get(j));
                j++;
            }
        }

        if (i == inorderSearch1.size()) {
            while (j < inorderSearch2.size()) {
                ansList.add(inorderSearch2.get(j++));
            }
        }

        if (j == inorderSearch2.size()) {
            while (i < inorderSearch1.size()) {
                ansList.add(inorderSearch1.get(i++));
            }
        }
        return ansList;

    }

    //中序遍历
    private List<Integer> inorderSearch(TreeNode root, ArrayList<Integer> nums) {
        if (root == null)
            return nums;
        inorderSearch(root.left, nums);
        nums.add(root.val);
        inorderSearch(root.right, nums);
        return nums;

    }


}

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;
    }
}