Python 实现把两个排好序的的列表合并成一个排序列表

Posted 2020-10-26 潇湘旧友

列表是生序的

# -*- coding: utf-8 -*-

# 合并两个排序的数组


def merge_list(a, b):
    if not a:
        return b
    if not b:
        return a
    a_index = b_index = 0
    ret = []
    while a_index < len(a) and b_index < len(b):
        if a[a_index] <= b[b_index]:
            ret.append(a[a_index])
            a_index += 1
        else:
            ret.append(b[b_index])
            b_index += 1
    if a_index < len(a):
        ret.extend(a[a_index:])
    if b_index < len(b):
        ret.extend(b[b_index:])
    return ret


if __name__ == ‘__main__‘:
    a = [1, 2, 3, 5, 7, 9]
    b = [1, 2, 2, 4, 5, 6, 8, 10]
    print merge_list(a, b)