两棵二叉搜索树中的所有元素
难度:中等
通过dfs分别遍历两棵树得到两个list集合,再将两个list集合合并排序即可。
代码如下:
public List<Integer> getAllElements(TreeNode root1, TreeNode root2) {
List<Integer> list1 = new ArrayList<>();
List<Integer> list2 = new ArrayList<>();
List<Integer> res = new ArrayList<>();
dfs(root1,list1);
dfs(root2,list2);
list1.addAll(list2);
Object[] objs = list1.toArray();
Arrays.sort(objs);
for (Object obj : objs) {
res.add((Integer) obj);
}
return res;
}
void dfs(TreeNode root, List<Integer> list) {
if (root == null) return ;
dfs(root.left, list);
list.add(root.val);
dfs(root.right, list);
}
执行结果:成功