插入排序——Python实现

插入排序Python实现

# -*- coding: utf-8 -*-
# @Time    : 2019/10/28 20:47
# @Author  : yuzhou_1shu
# @Email   : [email protected]
# @File    : insertion_sort.py
# @Software: PyCharm

def insertion_sort(collection):
    """Python实现插入排序算法
    将插入排序模拟成抓扑克牌：
    1、当抓到第一个牌肯定是有序的
    2、然后抓到第二张牌，如果比第一张大，放在右边；如果小，放在第一张左边；相等，左右都可以
    :param collection: 待排序序列
    :return: 升序排好的对应序列
    """

    length = len(collection)
    # 第一张牌肯定有序，所以循环变量从1开始（也就是第二张牌开始插入），i代表摸到的牌
    for i in range(1, length):
        insertion_index = i
        while(insertion_index > 0 and collection[insertion_index - 1] > collection[insertion_index]):
            collection[insertion_index], collection[insertion_index-1] = (collection[insertion_index-1], collection[insertion_index])
            insertion_index -= 1

    return collection


if __name__ == "__main__":

    list1 = [5, 5, 1, 4, 1, 2, 3]
    print("排序前: ", list1)
    print("排序后: ", insertion_sort(list1))