合并两个有序数组的思路

题目

https://leetcode.cn/problems/merge-sorted-array/description/

思路

双指针法(从后往前)

  1. 初始化两个指针p1和p2,分别指向nums1的前m个有效元素的末尾(即m - 1)和nums2的末尾(即n - 1)。同时,初始化一个指针p,指向合并后数组的末尾(即m + n - 1)。

  2. 比较nums1[p1]和nums2[p2]的值:
    如果nums1[p1] >= nums2[p2],则将nums1[p1]放到nums1[p]的位置,然后p1减 1,p减 1。
    如果nums1[p1] < nums2[p2],则将nums2[p2]放到nums1[p]的位置,然后p2减 1,p减 1。

  3. 重复上述比较和赋值操作,直到p2小于 0 或者p1小于 0 。

如果p2大于等于 0,说明nums2中还有剩余元素,将这些剩余元素依次复制到nums1的前面部分(从索引 0 开始)。因为nums1前面部分的m个元素在比较过程中可能已经被后移,所以此时nums1前面部分有足够空间容纳nums2的剩余元素。

而如果p1大于等于 0,nums1中剩余元素已经在正确位置,无需额外处理。

双指针法(从后往前)在实际操作中更为高效,因为它避免了在nums1前面部分频繁移动元素,时间复杂度为O(m + n),满足题目进阶要求。

image

Java 实现

public class MergeSortedArray {
    public void merge(int[] nums1, int m, int[] nums2, int n) {
        int p1 = m - 1;
        int p2 = n - 1;
        int p = m + n - 1;

        // 比较并合并
        while (p1 >= 0 && p2 >= 0) {
            if (nums1[p1] >= nums2[p2]) {
                nums1[p] = nums1[p1];
                p1--;
            } else {
                nums1[p] = nums2[p2];
                p2--;
            }
            p--;
        }

        // 如果nums2还有剩余元素,将其复制到nums1的前面部分
        System.arraycopy(nums2, 0, nums1, 0, p2 + 1);
    }

    public static void main(String[] args) {
        MergeSortedArray solution = new MergeSortedArray();
        int[] nums1 = {1, 2, 3, 0, 0, 0};
        int m = 3;
        int[] nums2 = {2, 5, 6};
        int n = 3;
        solution.merge(nums1, m, nums2, n);
        for (int num : nums1) {
            System.out.print(num + " ");
        }
    }
}

来自AI,未验证

posted @ 2025-05-08 15:05  talentzemin  阅读(45)  评论(0)    收藏  举报