LintCode Q6 Merge Two Sorted Arrays in Python

  • Jinhai ZHOU
  • 2 Minutes
  • 2016年12月11日
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
class Solution:
#@param A and B: sorted integer array A and B.
#@return: A new sorted integer array
def mergeSortedArray(self, A, B):
# write your code here
new = []
m = len(A)
n = len(B)
count = m + n
i, j = 0, 0
while i < m and j < n:
if A[i] <= B[j]:
new.append(A[i])
i += 1
else:
new.append(B[j])
j += 1
if i == m:
new.extend(B[j:])
else:
new.extend(A[i:])
return new
知识共享许可协议
本作品采用知识共享署名-相同方式共享 4.0 国际许可协议进行许可。