程序员最近都爱上了这个网站  程序员们快来瞅瞅吧!  it98k网:it98k.com

本站消息

站长简介/公众号

  出租广告位,需要合作请联系站长

+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

2023-05(3)

2023-06(2)

LeetCode_88、合并两个数组(python)

发布于2019-12-07 22:48     阅读(1163)     评论(0)     点赞(18)     收藏(2)


题目:

  • 给定两个有序整数数组 nums1 和 nums2,
  • 将 nums2 合并到 nums1 中,使得 num1 成为一个有序数组
  • 初始化 nums1 和 nums2 的元素数量分别为 m 和 n。
  • 假设 nums1 有足够的空间来保存 nums2 中的元素。

例子:

输入:
nums1 = [1,2,3],       m  = 3
nums2 = [2,5,6],       n = 3

输出: [1,2,2,3,5,6]
  • 1
  • 2
  • 3
  • 4
  • 5

思路:

双指针,从前往后遍历:

  • 定义指针p1指向数组nums1的开头
  • 定义指针p2指向数组nums2的开头
  • 不断的比较两个指针对应的元素,将小的元素放入结果数组中

所以最终的时间复杂度是 O(m + n) ,空间复杂度是O(m + n)

代码:

class Solution:
    def merge(self, nums1, m, nums2, n):
        nums1_copy = nums1[:m]            # 将数组nums1复制到数组nums1_copy
        nums1[:] = []                             # 清空数组nums1,用来存放结果
        
        p1 = 0                # 定义数组nums1_copy指针
        p2 = 0               # 定义数组nums2指针

        while p1 < m and p2 < n:             # 任意数组循环遍历完,就跳出
            if nums1_copy[p1] < nums2[p2]:          # 如果nums1_copy的元素小于nums2的元素
                nums1.append(nums1_copy[p1])      # 将小的元素添加到结果数组中
                p1 += 1                    # 指针右移,以便再次比较
            else:
                nums1.append(nums2[p2])                  # 将小的元素添加到结果数组中
                p2 += 1                    # 指针右移,以便再次比较
                
        if p1 < m:            # 如果数组nums1没有遍历完,说明剩下的都是大数,我们直接添加到结果数组中
            nums1[p1+p2:] = nums1_copy[p1:]
        if p2 < n:            # 如果数组nums2没有遍历完,说明剩下的都是大数,我们直接添加到结果数组中
            nums1[p1 + p2:] = nums2[p2:]
        return nums1          # 返回结果
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21

验证:

solution = Solution()
nums1 = [1,2,3,0,0,0]
nums2 = [2,5,6]
m = 3
n = 3
print(solution.merge(nums1,m,nums2,n))

# 结果:
[1, 2, 2, 3, 5, 6]
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9


所属网站分类: 技术文章 > 博客

作者:heer

链接:https://www.pythonheidong.com/blog/article/170247/bb7456f95e272b7454af/

来源:python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

18 0
收藏该文
已收藏

评论内容:(最多支持255个字符)