问题描述:
给你 root1 和 root2 这两棵二叉搜索树。请你返回一个列表,其中包含 两棵树 中的所有整数并按 升序 排序。.
样例如下:
代码如下:中序遍历+归并思路即可
import java.util.LinkedList;
import java.util.List;
public class GetAllElements {
//两颗二叉搜索树中所有元素
//给你 root1 和 root2 这两棵二叉搜索树。
// 请你返回一个列表,其中包含 两棵树 中的所有整数并按 升序 排序。.
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;
}
}
//前序遍历
public void preOrder(TreeNode root,List<Integer> list){
if (root!=null){
preOrder(root.left,list);
list.add(root.val);
preOrder(root.right,list);
}
}
public List<Integer> getAllElements(TreeNode root1, TreeNode root2) {//归并排序的思路
List<Integer> list1=new LinkedList<>();
List<Integer> list2=new LinkedList<>();
List<Integer> res=new LinkedList<>();
preOrder(root1,list1);
preOrder(root2,list2);
int st=0;
int end=0;
while (st<list1.size()&&end<list2.size()){
if (list1.get(st)<=list2.get(end)){
res.add(list1.get(st));
st++;
}else{
res.add(list2.get(end));
end++;
}
}
while (st<list1.size()){
res.add(list1.get(st));
st++;
}
while (end<list2.size()){
res.add(list2.get(end));
end++;
}
return res;
}
}