2022 把 Python 学得跟 Java 一样熟练——05 插入排序

背景

翻到以前实现的一个 Java 插入排序算法,今天把它翻译成 Python ,练习并温故 Python 的基础语法:

  1. for in range 循环下标范围,从 0 开始,到 N-1 ,倒序遍历用 reversed
  2. 插入排序算法,找到当前元素应该插入的位置。

回忆 Java 的实现

Java 实现插入排序算法:

public class InsertSortUtil {
    public static void insertSort(int[] array) {
        // 插入排序:i从第二个元素开始,将它与a0..ai-1 这些元素比较,找到它应该插入的位置 position
        for (int i = 1; i < array.length; i++) {
            // 记录当前 i 的元素到临时变量
            int currentValue = array[i];

            // 当前记录所在的位置为 i
            int position = i;

            // 向前对比,找到当前元素确定的有序的位置
            for (int j = i - 1; j >= 0; j--) {
                // 如果某个位置的元素比当前元素大,那就将这个数据向后挪动,修正位置
                if (array[j] > currentValue) {
                    array[j + 1] = array[j];
                    position = j;
                } else {
                    break;
                }
            }

            // 将目标位置插入当前元素
            array[position] = currentValue;
        }
    }

    public static void main(String[] args) {
        int[] r = new int[20];
        for (int i = 0; i < r.length; i++) {
            r[i] = (int) (Math.random() * 20000);
        }

        System.out.println("before sort:"+Arrays.toString(r));
        insertSort(r);
        System.out.println("after sort:"+Arrays.toString(r));
    }
}

Python 实现插入排序

def insertSort(array):
	for i in range(1,len(array)):
		# 记录临时变量
		currentValue = array[i]
		position = i
		
		for j in reversed(range(i)):
			# 如果当前 j 位置的数值比当前值大,将其往后移动,修正当前记录的位置
			if (array[j] > currentValue):
				array[j+1] = array[j]
				position = j
			else:
				break
		
		# 插入当前元素
		array[position] = currentValue

# 测试
datas = [1,2,0,3,4,6,7,9]
print('before sort',datas)		
insertSort(datas)
print('after sort',datas)

插入排序的核心思想是将当前元素 a[i] 插入到前面 a[0..i-1] 中某个合适的位置,内层遍历是逆序范围为【i-1,0】,使用 reversed 函数来翻转数组。

运行结果:
2022 把 Python 学得跟 Java 一样熟练——05 插入排序

启示录

消除模糊,今天弄透了两个 Python 知识点:

  1. 字符串的分片语法【start:end:step】,含头不含尾,这点跟 Java 语法一样。
  2. range(n) 得到的数组范围为 【0,n-1】,reversed 可以翻转数组。
上一篇:9-2·100 css下拉菜单


下一篇:吸顶效果方案