解题思路: 三种办法:1. 开辟一个新数组,空间复杂度O(n)。 2. 把B数组添加到A数组中,排序。 3. 类似于合并两个有序链表,设置两个指针,倒序插入元素
- 一共三个指针(这里的指针并不是真正的指向地址的指针,而是表示数组下标的风向标),a用来指数组A中的元素,b用来指数组B中的元素,total用来指向最终的m+n的A数组
- 控制a和b必须大于0,所以第一个while要求a大于0且b也大于0
- 后面两个while就好理解了,第一个while不满足的时候,要么A数组插完了,要么B数组插完了,就看后面两个while了
#
# @param A int整型一维数组
# @param B int整型一维数组
# @return void
#
class Solution:
def merge(self , A, m, B, n):
# write code here
a = m - 1
b = n - 1
total = m + n - 1
if n == 0:
return A
while a >= 0 and b >= 0:
if A[a] > B[b]:
A[total] = A[a]
total -= 1
a -= 1
else:
A[total] = B[b]
total -= 1
b -= 1
while a >= 0:
A[total] = A[a]
total -= 1
a -= 1
while b >= 0:
A[total] = B[b]
total -= 1
b -= 1
return A