LeetCode Easy: 88. Merge Sorted Array

一、题目

Given two sorted integer arrays nums1 and nums2, merge nums2 into nums1 as one sorted array.

Note:
You may assume that nums1 has enough space (size that is greater or equal to m + n) to hold additional elements from nums2. The number of elements initialized in nums1 and nums2 are m and n respectively.

 这道题虽然从题意表面上看是两个有序数组合并，但是按照合并数组的思路并不能AC，它的原意是选取nums2 的前 n 项 与 nums1的前 m 项合并，合并之后的nums1仍然是有序的。

二、思路

思路一、很简单，利用python的特性，将nums2的前n项，给nums1的m项之后做延伸，然后对nums1做sort（）即可，代码示例为 ：merge2；

思路二、从后往前遍历，取三个游标，其中两个游标分别对应nums1和nums2，因为nums1的长度不限，最终合并的数组依然在nums1中，所以设置一个新的合并之后的nums1的游标，长度设置为 m+n-1，然后nums1、nums2都是从后往前遍历，选择大的从nums1的m+n-1位置开始往前放，其中一个特殊情况是 n 遍历完了，直接将nums1的剩余的元素补充到前面就行了。

三、代码

merge1是我一开始没有理解题意，写的通过二分查找来合并数组的程序，仅供参考，并不能AC，merge2感觉像作弊，merge3即是思路2的实现。

#coding:utf-8
def merge1(nums1, nums2):
    """
    :type nums1: List[int]
    :type m: int
    :type nums2: List[int]
    :type n: int
    :rtype: void Do not return anything, modify nums1 in-place instead.
    """
    nums1 = list(nums1)
    nums2 = list(nums2)
    for cur1 in range(len(nums2)):
        left = 1
        right = len(nums1)-1
        while left <= right:
            mid = (left+right)//2
            # if left == right:
            #     nums1.insert(mid,nums2[cur1])
            if nums2[cur1] == nums1[mid]:
                nums1.insert(mid,nums2[cur1])
            elif nums2[cur1] > nums1[mid]:
                left = mid + 1
            else:
                #nums2[cur1] < nums1[mid]:
                right = mid - 1
        nums1.insert(mid,nums2[cur1])
    print(nums1)
    return nums1

def merge2(nums1, m, nums2, n):
    nums1[m:] = nums2[:n]
    nums1.sort()
    return nums1

def merge3(nums1, m, nums2, n):
    cur1 = m -1
    cur2 = n -1
    cur = cur1 + cur2 + 1  #cur is the index
    while cur1 >= 0 and cur2 >= 0:
        if nums1[cur1] > nums2[cur2]:
            nums1[cur] = nums1[cur1]
            cur1-=1
            cur-=1
        else:
            nums1[cur] = nums2[cur2]
            cur2-=1
        # if cur1 == 0:
        #     while cur2 != 0:
        #         nums1[cur2] = nums2[cur2]
        #         cur2 -= 1
        if cur2 > 0:
            while cur1 != 0:
                nums1[cur1] = nums1[cur1]
                cur1-=1

    print(nums1)
    return nums1

def merge4(nums1, m, nums2, n):
    i = m - 1
    j = n - 1
    k = m + n - 1
    while k >= 0:
        if (j < 0 or nums1[i] > nums2[j]) and i >= 0:
            nums1[k] = nums1[i]
            i -= 1
        else:
            nums1[k] = nums2[j]
            j -= 1
        k -= 1
    print(nums1)
    return nums1

if __name__ == '__main__':
    list1=[1,3,4,7,8,9]
    list2=[2,5,6]
    merge4(list1,5,list2,1)