难度:简单
类型:数组、双指针
题目:给定两个排序后的数组 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]
思路1:双指针分别指向数组的末端,即A[m-1]和B[n-1],逐一比较大小,较大的值拷贝到A的末尾指针前移
(仅适用于此处,A、B均有序)
思路2:拼接A与B,再对A排序sort
解1:
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_i = m - 1
B_i = n - 1
I_i = m + n -1
while B_i >= 0:
if A_i < 0:
A[0: B_i +1] = B[0: B_i +1]
break
if A[A_i] <= B[B_i]:
A[I_i] = B[B_i]
B_i -= 1
elif A[A_i] > B[B_i]:
A[I_i] = A[A_i]
A_i -= 1
I_i -= 1
执行用时 :28 ms, 在所有 Python3 提交中击败了99.22%的用户
内存消耗 :13.4 MB, 在所有 Python3 提交中击败了100.00%的用户
解2:
class Solution:
def merge(self, A: List[int], m: int, B: List[int], n: int) -> None:
A[m:]=B
A = A.sort()
执行用时 :48 ms, 在所有 Python3 提交中击败了31.78%的用户
内存消耗 :13.4 MB, 在所有 Python3 提交中击败了100.00%的用户
tips: LZ网速不稳定,相同代码执行用时有差别,此处两个解法用时不同不知道是算法问题还网速的锅!
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sorted-merge-lcci