数据结构(一)--合并两个数组为有序数组

    写在开头,最近复习数据结构。想来把数据结构中的算法或者代码都用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))

猜你喜欢

转载自blog.csdn.net/cyjmosthandsome/article/details/89389673