0
点赞
收藏
分享

微信扫一扫

两颗二叉搜索树中所有元素(java)

半秋L 2022-05-01 阅读 46

问题描述:
给你 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;
    }
}

在这里插入图片描述

举报

相关推荐

0 条评论