两棵二叉搜索树中的所有元素
- 题目描述
- 中序遍历 + 归并排序
题目描述
难度 - 中等
LC1305. 两棵二叉搜索树中的所有元素
给你 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^5 <= Node.val <= 10^5
中序遍历 + 归并排序
1.什么是搜索二叉树
二叉搜索树是一种计算机术语,也称为二叉查找树或二叉排序树。它或者是一棵空树,或者是具有以下性质的二叉树:
若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值;
若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值;
它的左、右子树也分别为二叉排序树。
根据二叉树的性质,可以得出,中序遍历的顺序就是递增的顺序,因此,我们可以用中序遍历把两个树,先遍历出来,放进集合中,两个集合就可以用归并排序进行排序,
代码演示:
/**
* 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 {
int INF = 0x3f3f3f3f;
public List<Integer> getAllElements(TreeNode root1, TreeNode root2) {
List<Integer> ans = new ArrayList<>();
List<Integer> l1 = new ArrayList<>(), l2 = new ArrayList<>();
dfs(root1, l1); dfs(root2, l2);
int n = l1.size(), m = l2.size(), i = 0, j = 0;
while (i < n || j < m) {
int a = i < n ? l1.get(i) : INF, b = j < m ? l2.get(j) : INF;
if (a <= b) {
ans.add(a); i++;
} else {
ans.add(b); j++;
}
}
return ans;
}
void dfs(TreeNode root, List<Integer> list) {
if (root == null) return ;
dfs(root.left, list);
list.add(root.val);
dfs(root.right, list);
}
}