合并两个有序数组

描述:

  给定两个有序整数数组 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]

思路:

这道题有两种做法,

    1. 一种是把两个数组合并,然后用排序的思路解决,但是这样做显然没有考虑到数据本身的规律
    2. 另一种是采用双指针的方式,利用数据的规律(有序)进行比较。时间复杂度是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]

结果:

 

 

 

 

posted @ 2019-12-02 13:33  进击的李同学  阅读(1043)  评论(0编辑  收藏  举报