一、题目链接
二、题目描述
给你 root1 和 root2 这两棵二叉搜索树。请你返回一个列表,其中包含 两棵树 中的所有整数并按 升序 排序。
三、解题思路
本题可以使用bfs+排序来解决。首先用bfs遍历两棵二叉树,并将其填入数组中,最后对数组进行排序即可。
四、源码
class Solution {
public:
vector<int> vec;
void dfs(TreeNode* root){
if(root!=NULL){
if(root->left!=NULL){
dfs(root->left);
}
vec.push_back(root->val);
if(root->right!=NULL){
dfs(root->right);
}
}
}
vector<int> getAllElements(TreeNode* root1, TreeNode* root2) {
dfs(root1);
dfs(root2);
sort(vec.begin(),vec.end());
return vec;
}
};