合并两个有序数组
描述:
给定两个有序整数数组 nums1 和 nums2,将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组。
说明:
初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。
示例:
输入:
nums1 = [1,2,3,0,0,0], m = 3
nums2 = [2,5,6], n = 3
输出: [1,2,2,3,5,6]
思路:
这道题有两种做法,
- 一种是把两个数组合并,然后用排序的思路解决,但是这样做显然没有考虑到数据本身的规律
- 另一种是采用双指针的方式,利用数据的规律(有序)进行比较。时间复杂度是O(n+m),空间复杂度为O(1),显然优于第一种做法
java(方法二):
class Solution { 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)) { nums1[p--] = (nums1[p1] < nums2[p2]) ? nums2[p2--] : nums1[p1--]; } System.arraycopy(nums2, 0, nums1, 0, p2 + 1); } }
结果:
python3(方法二):
class Solution: def merge(self, nums1: List[int], m: int, nums2: List[int], n: int) -> None: """ Do not return anything, modify nums1 in-place instead. """ # 初始化指针 p1 = m - 1 p2 = n - 1 p = m + n - 1 while p1 >= 0 and p2 >= 0: if nums1[p1] < nums2[p2]: nums1[p] = nums2[p2] p2 -= 1 else: nums1[p] = nums1[p1] p1 -= 1 p -= 1 nums1[:p2 + 1] = nums2[:p2 + 1]
结果: