leetcode--python--88

88. 合并两个有序数组

给你两个有序整数数组 nums1 和 nums2,请你将 nums2 合并到 nums1 中,使 nums1 成为一个有序数组。

说明:

初始化 nums1 和 nums2 的元素数量分别为 m 和 n 。
你可以假设 nums1 有足够的空间(空间大小大于或等于 m + n)来保存 nums2 中的元素。

这道题也是双指针的题,和之前多的这道题一模一样,所以代码也一样

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.
        """
        tem_list = nums1[:m] + nums2
        nums1[:] = []			#需要注意是这里,看清题目,返回的是nums1,所以这里赋值一下
        start_1, start_2 = 0,m
        while start_1 < m or start_2 < m+n:

            if start_1 == m:
                nums1.append(tem_list[start_2])
                start_2+=1
            elif start_2 == m+n:
                nums1.append(tem_list[start_1])
                start_1+=1
            elif tem_list[start_1] <= tem_list[start_2]:
                nums1.append(tem_list[start_1])
                start_1+=1
            else:
                nums1.append(tem_list[start_2])
                start_2+=1
       		

(写完之后感觉给搞复杂了),可以直接不用把2个数组相加的,可以减小空间复杂度

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.
        """
        tem = nums1[:m]
        nums1[:] = []
        i,j = 0, 0
        while i < m or j < n:
            if i == m:
                nums1.append(nums2[j])
                j+=1
                
            elif j == n:
                nums1.append(tem[i])
                i+=1
                
            elif tem[i] <= nums2[j]:
                nums1.append(tem[i])
                i+=1
               
            else:
                nums1.append(nums2[j])
                j+=1

猜你喜欢

转载自blog.csdn.net/AWhiteDongDong/article/details/111064644
今日推荐