题目:
描述
给出一个有序的整数数组 A 和有序的整数数组 B ,请将数组 B 合并到数组 A 中,变成一个有序的升序数组
数据范围: 0 \le n,m \le 1000≤n,m≤100,|A_i| <=100∣A i∣<=100, |B_i| <= 100∣B i∣<=100
注意:
1.保证 A 数组有足够的空间存放 B 数组的元素, A 和 B 中初始的元素数目分别为 m 和 n,A的数组空间大小为 m+n
2.不要返回合并的数组,将数组 B 的数据合并到 A 里面就好了,且后台会自动将合并后的数组 A 的内容打印出来,所以也不需要自己打印
3. A 数组在[0,m-1]的范围也是有序的
解题思路:
三种办法: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