LeetCode每日一题刷题记录1305.两颗二叉搜索树中的所有元素

这个题学过二叉搜索数的话就很简单，二叉搜索数有这么一个特性，左子树中所有元素小于根节点值，右子树中所有元素大于根节点，因此，对其中序遍历即可得到有序序列，这道题给了两颗树，对其分别中序遍历再将得到的两序列合并即可

class Solution {
public:
    vector<int> nums;
    void preoder(TreeNode* root)
    {
        if (root == nullptr)
            return;
        preoder(root->left);
        nums.push_back(root->val);
        preoder(root->right);
    }
    vector<int> getAllElements(TreeNode* root1, TreeNode* root2)
    {
        vector<int> nums1;
        preoder(root1);
        nums1 = nums;
        nums.clear();
        preoder(root2);
        vector<int> nums2 = nums;
        int len1 = nums1.size(), len2 = nums2.size();
        nums = vector<int>(len1 + len2);
        int i = 0, j = 0, k = 0;
        while (i < len1 && j < len2)
        {
            if (nums1[i] > nums2[j])
                nums[k++] = nums2[j++];
            else
            {
                nums[k++] = nums1[i++];
            }
        }
        while (i < len1)
        {
            nums[k++] = nums1[i++];
        }
        while (j < len2)
        {
            nums[k++] = nums2[j++];
        }
        return nums;
    }
};

当然，这道题除了这样做以外还有一个做法，可以通过层序遍历一次在一个数组获得所有元素然后再排序

public:
    vector<int> nums;
    vector<int> getAllElements(TreeNode* root1, TreeNode* root2)
    {
        queue<TreeNode*> temp;
        if(root1 != nullptr)
            temp.push(root1);
        if(root2 !=nullptr)
            temp.push(root2);
        TreeNode* top;
        while (!temp.empty())
        {
            top = temp.front();
            nums.push_back(top->val);
            if (top->left != nullptr)
                temp.push(top->left);
            if (top->right != nullptr)
                temp.push(top->right);
            temp.pop();
        }
        sort(nums.begin(), nums.end());
        return nums;
    }
};