首页 > 其他分享> 文章详细

leetcode 1305. 两棵二叉搜索树中的所有元素

2021-05-18 15:33:46 阅读：173 来源： 互联网

标签：root1 1305 TreeNode 10 List list 树中 leetcode root2

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

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

示例 1：

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

输入：root1 = [1,null,8], root2 = [8,1]
输出：[1,1,8,8]

提示：

每棵树最多有 5000 个节点。
每个节点的值在 [-10^5, 10^5] 之间。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/all-elements-in-two-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

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

        return sort(a, b);

    }


    private List<Integer> sort(List<Integer> a, List<Integer> b) {
        int s1 = a.size();
        int s2 = b.size();
        List<Integer> list = new ArrayList<>(s1 + s2);
        int i = 0;
        int j = 0;
        while (i < s1 && j < s2) {
            if (a.get(i) > b.get(j)) {
                list.add(b.get(j++));
            } else {
                list.add(a.get(i++));
            }
        }
        while (i < s1) {
            list.add(a.get(i++));
        }

        while (j < s2) {
            list.add(b.get(j++));
        }
        return list;
    }

    private void find(TreeNode node, List<Integer> list) {
        if (node == null) {
            return;
        }
        find(node.left, list);
        list.add(node.val);
        find(node.right, list);
    }
}

标签：root1,1305,TreeNode,10,List,list,树中,leetcode,root2
来源： https://www.cnblogs.com/wangzaiguli/p/14781101.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

leetcode 1305. 两棵二叉搜索树中的所有元素