LeetCode - #88 合并两个有序数组，真以为有序就可以放肆了？

IOS

2023-12-26 06:12:49

前言

在学习 LeetCode 算法题的过程中，相信很多人都遇到过 #88 合并两个有序数组这道题。乍一看，这道题似乎很容易，毕竟两个数组都是有序的，只需要将两个数组中的元素一一比较，然后将较小的元素添加到新的数组中即可。

然而，事情并没有那么简单。当两个数组的长度都非常大的时候，这种朴素的算法就会变得非常慢。为了提高算法的效率，我们需要使用一种更巧妙的方法来合并两个有序数组。

算法详解

在 LeetCode 上，这道题的官方解法是使用双指针法。双指针法是一种非常巧妙的算法，它利用了两个有序数组的特点，可以将时间复杂度降低到 O(n)，其中 n 是两个数组的总长度。

双指针法的工作原理如下：

将两个数组的指针都指向第一个元素。
比较两个指针指向的元素，并将较小的元素添加到新的数组中。
将较小的元素的指针向后移动一位。
重复步骤 2 和 3，直到两个指针都指向最后一个元素。

这样一来，我们就得到了一个新的有序数组，而时间复杂度只有 O(n)。

注意事项

在使用双指针法合并两个有序数组时，需要注意以下几点：

如果两个数组中包含重复元素，那么需要对两个数组进行去重处理，否则合并后的数组中会出现重复元素。
如果两个数组中包含负数元素，那么需要对两个数组进行排序，否则合并后的数组不会有序。
如果两个数组中包含空元素，那么需要对两个数组进行非空处理，否则合并后的数组中会出现空元素。

总结

LeetCode 上的 #88 合并两个有序数组题目看似简单，但实际上却隐藏着不少玄机。通过这篇文章，你已经了解了如何使用双指针法来合并两个有序数组，以及在使用双指针法时需要注意的事项。希望这篇文章能够帮助你更好地理解这道题，并提高你在 LeetCode 上的算法能力。

代码实现

func merge(_ nums1: inout [Int], _ m: Int, _ nums2: [Int], _ n: Int) {
    var i = m - 1
    var j = n - 1
    var k = m + n - 1

    while i >= 0 && j >= 0 {
        if nums1[i] > nums2[j] {
            nums1[k] = nums1[i]
            i -= 1
        } else {
            nums1[k] = nums2[j]
            j -= 1
        }
        k -= 1
    }

    while j >= 0 {
        nums1[k] = nums2[j]
        j -= 1
        k -= 1
    }
}