leetcode 88 - 合并两个有序数组的三种解法

题目: https://leetcode.cn/problems/merge-sorted-array/submissions/
都是官方的解答。照着答案领悟了一遍,现在觉得确实在那一堆坑里面算简单的题目。
程序代码结构如下,
合并有序数组结构图
CMakeLists.txt

cmake_minimum_required(VERSION 2.6)

project(merge_sorted_array)
set(CMAKE_CXX_STANDARD 20)
add_definitions(-g)

include_directories(${CMAKE_CURRENT_SOURCE_DIR}/../../../include)
string(REPLACE ".cpp" "" file "main.cpp")
add_executable(${file}  "main.cpp")

main.cpp

#include "printer/printer.hpp"

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

class Solution {
public:
    // 时间复杂度 (m+n)*log(m+n)
    // 空间复杂度 log(m + n)
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        nums1.erase(nums1.begin() + m, nums1.end());
        copy(nums2.begin(), nums2.end(), back_inserter(nums1));
        sort(nums1.begin(), nums1.end());
    }

    void merge_insert(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int p1 {0}, p2{0}, cur{0};
        vector<int> sorted(m+n, 0);
        for(int i=0; i<m+n; ++i) {
            // 说明nums1搞完了
            if(p1==m) {
                cur = nums2[p2++];
                // 说明 nums2搞完了
            } else if(p2==n) {
                cur = nums1[p1++];
                // 都没搞完,nums1比较小
            } else if(nums1[p1] < nums2[p2] ){
                cur = nums1[p1++];
            } else {
                cur = nums2[p2++];
            }
            sorted[p1+p2-1] = cur;
        }
        sorted.swap(nums1);
    } 

    void merge_insert2(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int p1{m-1}, p2{n-1};
        int tail {m+n-1};
        int cur{0};
        while(p1>=0 || p2>=0) {
            // 数组1 完事了
            if(p1 == -1) {
                cur = nums2[p2--];
                //数组2 完事了
            }else if(p2==-1) {
                cur = nums1[p1--];
                // 两个都还有,大的放后面
            } else if(nums1[p1] > nums2[p2]) {
                cur = nums1[p1--];
            } else {
                cur = nums2[p2--];
            }
            nums1[tail--] = cur;
        }
    }
};

auto main(int argc, char** argv) -> int {
    vector<int> nums1 {1,2,3,0,0,0};
    vector<int> nums2 {2,5,6};
    
    cout << "before merge: " << endl;
    cout << "nums1: " << nums1 << endl;
    cout << "nums2: " << nums2 << endl;

    Solution s;
    // s.merge(nums1, 3, nums2, nums2.size());
    // cout << "after merge, result is: " << nums1 << endl;

    // s.merge_insert(nums1, 3, nums2, nums2.size());
    // cout << "after merge_insert, result is: " << nums1 << endl;

    s.merge_insert2(nums1, 3, nums2, nums2.size());
    cout << "after merge_insert2, result is: " << nums1 << endl;

    return EXIT_SUCCESS;
}

程序输出如下
程序输出
leetcode上最后一种尾插法比较推荐。
输出如下,
leetcode示意图

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值