Leetcode之两棵二叉搜索树中的所有元素

问题描述#

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

问题解法#

中序遍历+合并排序#

这是一开始的想法。

Copy
/**
 * 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> list1 =new ArrayList<Integer>();
		  List<Integer> list2 =new ArrayList<Integer>();
		  List<Integer> list =new ArrayList<Integer>();
		  list1=minsearch(root1, list1);
		  list2=minsearch(root2, list2);
		  int i=0,j=0;
		  for(;i<list1.size()&&j<list2.size();) {
			  if(list1.get(i)<list2.get(j)) {
				  list.add(list1.get(i));
				  i++;
			  }else {
				  list.add(list2.get(j));
				  j++;
			  }
		  }
		  if(i>=list1.size()) {
			  for(;j<list2.size();j++) {
				  list.add(list2.get(j));
			  }
		  }
		  if(j>=list2.size()) {
			  for(;i<list1.size();i++) {
				  list.add(list1.get(i));
			  }
		  }
		  return list;
	    }
	  public  List<Integer> minsearch(TreeNode q,List<Integer> listp) {
		  if(q==null)
			  return listp;
			  minsearch(q.left,listp);
			  listp.add(q.val);
			  minsearch(q.right,listp);
		  return listp;
	  }
}

中序遍历到一个集合后，使用Collections来排序#

Copy
**
 * 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> list1 =new ArrayList<Integer>();
		  list1=minsearch(root1, list1);
           list1=minsearch(root2, list1);
		  Collections.sort(list1);
          return list1;
    }
 public  List<Integer> minsearch(TreeNode q,List<Integer> listp) {
		  if(q==null)
			  return listp;
			  minsearch(q.left,listp);
			  listp.add(q.val);
			  minsearch(q.right,listp);
		  return listp;
	  }

}