1. FlyPython首页
  2. 数据结构与算法
  3. leetcode题解

「LeetCode每日一题」——LCCI 10.01. 合并排序的数组

LCCI 10.01. 合并排序的数组

链接:https://leetcode-cn.com/problems/sorted-merge-lcci/
难度:简单

题目

给定两个排序后的数组 A 和 B,其中 A 的末端有足够的缓冲空间容纳 B。 编写一个方法,将 B 合并入 A 并排序。

初始化 A 和 B 的元素数量分别为 m 和 n。

示例:

输入:
A = [1,2,3,0,0,0], m = 3
B = [2,5,6], n = 3

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

思路

这个题的思路主要是有两个。第一种就是双指针,从后往前扫描,较大的往后放,直到A数组遍历完,然后数组B就直接复制到数组A。第二种就是加起来排序,当然Python写第二种更简单。具体代码请见下面方案代码。

方案代码

解法:

class Solution:
    def merge(self, A: List[int], m: int, B: List[int], n: int) -> None:
        """
        Do not return anything, modify A in-place instead.
        """
        A[m:] = B
        A.sort()
    

原创文章,作者:flypython,如若转载,请注明出处:http://flypython.com/algorithm/leetcode/162.html