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);
}