合并两个有序数组

题目

leetcode 88 简单

给你两个按 非递减顺序 排列的整数数组 nums1 和 nums2，另有两个整数 m 和 n ，分别表示 nums1 和 nums2 中的元素数目。

请你 合并 nums2 到 nums1 中，使合并后的数组同样按 非递减顺序 排列。

注意：最终，合并后数组不应由函数返回，而是存储在数组 nums1 中。为了应对这种情况，nums1 的初始长度为 m + n，其中前 m 个元素表示应合并的元素，后 n 个元素为 0 ，应忽略。nums2 的长度为 n 。

解析

对于普通的合并两个有序的数组的问题，我们可以使用i,j双指针分别放在两个有序数组的开头，对比两个指针指向元素的大小，将较小的元素放入到合并好的数组中，然后移动指针。最后将剩余的元素放入到合并好的数组。

题目要求函数不能返回数组，合并后的元素存储在 num1 中，只需再遍历一遍合并好的数组，将 nums1 中的元素覆盖即可。

js代码实现

function merge(nums1, m, nums2, n) {
    let p1 = 0, p2 = 0
    let sorted = new Array(m + n).fill(0)
    let cur
    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++]
      }
      sorted[p1 + p2 - 1] = cur
    }
    for (let i = 0; i < sorted.length; i++) {
      nums1[i] = sorted[i]
    }
}


let nums1 = [1,2,3,0,0,0], m = 3, nums2 = [2,5,6], n = 3

merge(nums1, m, nums2, n)

// 或者使用下面这种方法，内存占用稍微大一些
const mergeSortedArrays = function(nums1, m, nums2, n) {
    let p1 = m - 1; // nums1 的末尾指针
    let p2 = n - 1; // nums2 的末尾指针
    let 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 中
    while (p2 >= 0) {
        nums1[p] = nums2[p2];
        p2--;
        p--;
    }
};

go代码实现

package main

import (
	"fmt"
)

func mergeSortedArrays(nums1, nums2 []int, m, n int) {
	p1, p2 := 0, 0
	var cur int
	sortedArray := make([]int, m+n)
	for p1 < m || p2 < n {
		if p1 == m {
			cur = nums2[p2]
			p2++
		} else if p2 == n {
			cur = nums1[p1]
			p1++
		} else if nums1[p1] < nums2[p2] {
			cur = nums1[p1]
			p1++
		} else {
			cur = nums2[p2]
			p2++
		}
		sortedArray[p1+p2-1] = cur
	}
	for i, v := range sortedArray {
		nums1[i] = v
	}
}

func main() {
	nums1, m := []int{1, 2, 3, 0, 0, 0}, 3
	nums2, n := []int{2, 5, 6}, 3

	mergeSortedArrays(nums1, nums2, m, n)
	fmt.Println(nums1) // [1,2,2,3,5,6]
}