写在开头,最近复习数据结构。想来把数据结构中的算法或者代码都用python实现一下。
--------------------------------------------------------------------------分割线-----------------------------------------------------------------------------------------
描述:
有两个无序数组,A,B, 将两个数组合并为从小到大排序的有序数组。
如A=[3, 8, 11, 5], B = [2, 6, 11, 8, 9, 20, 15]
应该得到数组为:[2, 3, 5, 6, 8, 8, 9, 11, 11, 15, 20]
见《数据结构》P20-P21
算法思想:
将两个数组分别由小到大排序,用两个指针k,j(不需要指针实现,只是用于理解算法)分别
指向数组A和数组B的第一个元素,定义一个结果数组res。如果k指向的元素小于等于j指向的元素,则k加入res,
且k后移一个元素,否则j加入res,且j后移一个元素。
如果在循环中A已经遍历到末尾,而B还没有,则将B剩余的元素直接全部添加到res最后一个元素的后面。
对于B已经遍历完,而A还没有的情况,亦然。
python3实现:
class Solution:
def hebing(self, A, B):
A.sort()
B.sort()
res = []
while A or B:
if not A:
res.extend(B)
return res
elif not B:
res.extend(A)
return res
else:
res.append(A.pop(0) if A[0] <= B[0] else B.pop(0))
return res
if __name__ == '__main__':
A = [3, 8, 11, 5]
B = [2, 6, 11, 8, 9, 20, 15]
t = Solution()
print(t.hebing(A, B))