[路飞]_js算法:leetcode 1305-两棵二叉搜索树中的所有元素

74 阅读1分钟

leetcode 1305. 两棵二叉搜索树中的所有元素

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

思路: 采用归并排序的方法 将二叉搜索树中序遍历制成数组

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root1
 * @param {TreeNode} root2
 * @return {number[]}
 */
var getAllElements = function(root1, root2) {
   let num1=[],num2=[];
  middleOpera(root1,num1)
  middleOpera(root2,num2)
  let temp=[],p1=0,p2=0;
  while((p1<=num1.length-1)||(p2<=num2.length-1)){
    if(p2>num2.length-1||(p1<=num1.length-1&&num1[p1]<=num2[p2])){
      temp.push(num1[p1++])
    }else{
      temp.push(num2[p2++])
    }
  }
  return temp
};

function middleOpera(root,arr){
  if(root==null)return;
  middleOpera(root.left,arr)
  arr.push(root.val);
  middleOpera(root.right,arr);
}