python 算法——插入排序

Posted 2021-04-25 Python数据之路

插入排序

1. 插入排序（英语：Insertion Sort）是一种简单直观的排序算法。它的工作原理是通过构建有序序列，对于未排序数据，在已排序序列中从后向前扫描，找到相应位置并插入。插入排序在实现上，通常采用in-place排序（即只需用到  O(1) 的额外空间的排序），因而在从后向前扫描过程中，需要反复把已排序元素逐步向后挪位，为最新元素提供插入空间。

算法描述

1. 从第一个元素开始，该元素可以认为已经被排序

2. 取出下一个元素，在已经排序的元素序列中从后向前扫描

3. 如果该元素（已排序）大于新元素，将该元素移到下一位置

4. 重复步骤3，直到找到已排序的元素小于或者等于新元素的位置

5. 将新元素插入到该位置后

6. 重复步骤2~5
   

   插入排序

Python 代码实现

def insertion_sort(collection):
    for index in range(1, len(collection)):
        while index > 0 and collection[index] < collection[index - 1]:
            collection[index], collection[index - 1] = collection[index - 1], collection[index]
            index -= 1
    return collection
if __name__ == '__main__':
    user_input = input("请输入带逗号的数字（例如：1，2，3）：")
    unsorted = [int(item) for item in user_input.strip().split(',')]
    print(insertion_sort(unsorted))