python合并排序数组。 输入A = [1],B = [1],输出[1,1],返回合并后的数组。 输入A = [1,2,3,4],B = [2,4,5,6],输出[1,2,2,3,4,4,5,6],返回合并所有元素后的数组
时间: 2023-07-02 07:09:58 浏览: 91
python实现合并两个数组的方法
5星 · 资源好评率100%
可以使用归并排序的思想来解决这个问题。具体实现步骤如下:
1.定义一个空列表result用于存放最终合并后的结果。
2.定义两个指针i和j,初始值分别为0,指向数组A和B的第一个元素。
3.比较A[i]和B[j]的大小,将较小的值添加到result中,并将对应的指针向后移动一位。
4.重复步骤3,直到其中一个指针超出了数组范围。
5.将剩余的元素添加到result中。
6.返回result。
代码实现如下:
```python
def merge_sorted_array(A, B):
result = []
i, j = 0, 0
while i < len(A) and j < len(B):
if A[i] <= B[j]:
result.append(A[i])
i += 1
else:
result.append(B[j])
j += 1
result += A[i:]
result += B[j:]
return result
```
测试代码:
```python
A = [1]
B = [1]
print(merge_sorted_array(A, B)) # [1, 1]
A = [1, 2, 3, 4]
B = [2, 4, 5, 6]
print(merge_sorted_array(A, B)) # [1, 2, 2, 3, 4, 4, 5, 6]
```
运行结果:
```
[1, 1]
[1, 2, 2, 3, 4, 4, 5, 6]
```
阅读全文