1.题目
给你 root1 和 root2 这两棵二叉搜索树。请你返回一个列表,其中包含 两棵树 中的所有整数并按 升序 排序。.
2.示例
示例 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]
提示:
每棵树的节点数在 [0, 5000] 范围内
-105 <= Node.val <= 105
3.思路
用二叉树,升序选择前序遍历,两个树的话可以分别排序存数组,再比较两个数组的最小值,将最小值依次加入答案数组
4.代码
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
//将二叉树中序遍历,获得排序好的数组
func inorder(root *TreeNode) (res []int) {
var dfs func(*TreeNode)
dfs = func(node *TreeNode) {
if node == nil {
return
}
dfs(node.Left)
res = append(res, node.Val)
dfs(node.Right)
}
dfs(root)
return
}
func getAllElements(root1, root2 *TreeNode) []int {
//用上面的方法得到两个排序后的数组
nums1 := inorder(root1)
nums2 := inorder(root2)
p1, n1 := 0, len(nums1)
p2, n2 := 0, len(nums2)
//遍历两个数组,按顺序合成一个数组
ans := make([]int, 0, n1+n2)
for {
//如果nums1的数据已经全部加入ans数组,则把nums2剩余的数字加入ans
//数组组合完成,返回答案
if p1 == n1 {
return append(ans, nums2[p2:]...)
}
if p2 == n2 {
return append(ans, nums1[p1:]...)
}
if nums1[p1] < nums2[p2] {
ans = append(ans, nums1[p1])
p1++
} else {
ans = append(ans, nums2[p2])
p2++
}
}
}