leetcode_88. Merge Sorted Array 合并两个排好序的数组,尾插法,从尾数开始合并

来源:互联网 发布:linux下语言包安装 编辑:程序博客网 时间:2024/06/03 19:20

题目:

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.


题意:

给定两个排好序的数组nums1和nums2,将nums2合并到nums1。假设nums1空间足够大,不用返回nums1,直接修改nums1。


代码:

class Solution(object):
    def merge(self, nums1, m, nums2, n):
        """
        :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.
        """
        
        '''
        if n > 0 :
            nums1[m:] = nums2[:n]                            #方法1:将nums2加在nums1后面,再对nums1排序就可以了
            nums1.sort()
        '''
        
        if n > 0 :                           #方法2:尾插法,从nums1和nums2的末尾开始比较来排序,依次取nums1和nums2末尾中最大的一个,从nums1的末尾开始插起,不用移动元素
            k = m+n-1
            i = m-1
            j = n-1
            while i >= 0 and j >= 0 :
                if nums1[i] >= nums2[j] :
                    nums1[k] = nums1[i]
                    i -= 1
                else :
                    nums1[k] = nums2[j]
                    j -= 1
                k -= 1
            
            if i >= 0 :                       #对剩余不用比较的元素进行处理
                while i >= 0 :
                    nums1[k] = nums1[i]
                    i -= 1
                    k -= 1
            if j >= 0 :                     #对剩余不用比较的元素进行处理
                while j >= 0 :
                    nums1[k] = nums2[j]
                    j -= 1
                    k -= 1
                
                

笔记:

尾插法,有印象,但是当时忘记了。温故而知新啊!


0 0
原创粉丝点击