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

最新推荐文章于 2024-09-26 09:13:29 发布

sakeww

最新推荐文章于 2024-09-26 09:13:29 发布

阅读量238

点赞数 12

分类专栏： leetcode，牛客日常

本文链接：https://blog.csdn.net/sakeww/article/details/124522541

版权

c++ leetcode 深度优先

leetcode，牛客日常专栏收录该内容

113 篇文章 3 订阅

订阅专栏

链接：

https://leetcode-cn.com/problems/all-elements-in-two-binary-search-trees/

描述和示例：

在这里插入图片描述

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
   void middle(TreeNode* root,vector<int>& arr)
   {
        if(root){
           middle(root->left,arr);
           arr.push_back(root->val);
           middle(root->right,arr);
        }
       
   }

    vector<int> getAllElements(TreeNode* root1, TreeNode* root2) {
        vector<int> arr1,arr2;
        middle(root1,arr1);
        middle(root2,arr2);

        vector<int> ret;
        auto p1=arr1.begin();
        auto p2=arr2.begin();

        while(true)
        {
            if(p1==arr1.end())
            {
                ret.insert(ret.end(),p2,arr2.end());
                break;
            }
            if(p2==arr2.end())
            {
                ret.insert(ret.end(),p1,arr1.end());
                break;
            }

            if(*p2>*p1)
                ret.push_back(*p1++);
            else 
                ret.push_back(*p2++);
        }
        return ret;
    }
};