从今天开始重刷力扣 Leetcode 88. 合并两个有序数组双指针

自信的小螺丝钉

已于 2024-05-17 14:43:36 修改

阅读量273

点赞数 9

分类专栏： Leetcode 文章标签： leetcode 算法双指针 C++ 刷题

于 2024-05-17 14:42:07 首次发布

本文链接：https://blog.csdn.net/qq_45791939/article/details/139003778

版权

Leetcode 专栏收录该内容

324 篇文章 0 订阅

订阅专栏

Leetcode 88. 合并两个有序数组

在这里插入图片描述

官方要求时间复杂度O(m+n)，要用双指针完成，将空间复杂度降为O(1)的逆向双指针值得学习

1. 我最开始冗长的代码：

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        if (m == 0)
            nums1.assign(nums2.begin(), nums2.end());
        if (m != 0 && n != 0) {
            int nums1_min = nums1[0];
            int nums1_max = nums1[m - 1];
            int nums2_min = nums2[0];
            int nums2_max = nums2[n - 1];
            if (nums2_min >= nums1_max) {
                for (int i = m; i < m + n; i++) {
                    nums1[i] = nums2[i - m];
                }
            } else if (nums2_max <= nums1_min) {
                vector<int> tmp(nums1);
                for (int i = 0; i < n; i++) {
                    nums1[i] = nums2[i];
                }
                for (int i = n; i < m + n; i++) {
                    nums1[i] = tmp[i - n];
                }
            } else {
                vector<int> tmp(nums1);
                int f1 = 0, f2 = 0, i = 0;
                int x = tmp[0], y = nums2[0];
                while (f1 < m && f2 < n) {
                    x = tmp[f1], y = nums2[f2];
                    if (x <= y) {
                        nums1[i] = x;
                        f1++;
                    } else {
                        nums1[i] = y;
                        f2++;
                    }
                    i++;
                }
                while (f1 < m) {
                    x = tmp[f1];
                    nums1[i] = x;
                    f1++;
                    i++;
                }
                while (f2 < n) {
                    y = nums2[f2];
                    nums1[i] = y;
                    f2++;
                    i++;
                }
            }
        }
    }
};

2. 看了官解之后精炼的代码，同样是双指针，空间复杂度O(m+n)

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int p1 = 0, p2 = 0;
        int cur;
        vector<int> tmp;
        while (p1 < m || p2 < n) {
            if (p1 == m) {
                cur = nums2[p2++];
            } else if (p2 == n) {
                cur = nums1[p1++];
            } else if (nums1[p1] <= nums2[p2]) {
                cur = nums1[p1++];
            } else {
                cur = nums2[p2++];
            }
            tmp.push_back(cur);
        }
        for (int i = 0; i < m + n; i++) {
            nums1[i] = tmp[i];
        }
    }
};

3. 逆向双指针，空间复杂度O(1)

class Solution {
public:
    void merge(vector<int>& nums1, int m, vector<int>& nums2, int n) {
        int p1 = m - 1, p2 = n - 1, i = m + n - 1;
        while (p1 >= 0 || p2 >= 0) {
            if (p1 < 0) {
                nums1[i--] = nums2[p2--];
            } else if (p2 < 0) {
                nums1[i--] = nums1[p1--];
            } else if (nums1[p1] >= nums2[p2]) {
                nums1[i--] = nums1[p1--];
            } else {
                nums1[i--] = nums2[p2--];
            }
        }
    }
};