LeetCode 5.1每日一题

一、题目链接

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

二、题目描述

        给你 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;
    }
};