春招第一步,算法伴我行
两个有序数组的合并是一道很简单的题目,方法也有很多。我自己的思路还是偏向归并排序,即使用两个指针,分别遍历两个排序数组,遇到较小的,放入结果数组中,对应指针加1,直到其中一个遍历结束。此时需要进行指针位置判断,这种情况是处理其中一个没有遍历完成,即剩余的元素都比另外一个数组大,需要单独处理:
if i<len(A): result = result+A[i:]; if j<len(B): result = result+B[j:]
完整代码如下:
class Solution():
def MergeTwoList(self, A, B):
if not A:
return B
if not B:
return A
i,j = 0, 0
result = []
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
if i<len(A):
result+=A[i:]
if j<len(B):
result+=B[j:]
return result```