📜  python中的排序算法(1)

📅  最后修改于: 2023-12-03 14:46:40.730000             🧑  作者: Mango

Python中的排序算法

排序算法是计算机科学中的一个重要主题。在Python中,有多种已实现的排序算法,包括冒泡排序、选择排序、插入排序、快速排序、归并排序、堆排序等等。不同的排序算法对于不同的数据集都有其优缺点,程序员可以根据实际情况来选择合适的排序算法。

冒泡排序

冒泡排序是一种简单的排序算法,它的原理是比较相邻的两个元素,如果前一个元素大于后一个元素,则交换它们。这样一轮下来,最大的元素就会被交换到最后面。重复这个过程,直到整个序列都变成有序的。

def bubble_sort(arr):
    n = len(arr)
    for i in range(n - 1):
        for j in range(0, n-i-1):
            if arr[j] > arr[j+1]:
                arr[j], arr[j+1] = arr[j+1], arr[j]
    return arr

时间复杂度:O(n^2)

选择排序

选择排序的原理是从待排序序列中选择最小的元素放在起始位置,再从剩余的元素中选择最小的元素放在已排序序列的末尾。重复这个过程,直到整个序列都变成有序的。

def selection_sort(arr):
    n = len(arr)
    for i in range(n - 1):
        min_idx = i
        for j in range(i+1, n):
            if arr[j] < arr[min_idx]:
                min_idx = j
        arr[i], arr[min_idx] = arr[min_idx], arr[i]
    return arr

时间复杂度:O(n^2)

插入排序

插入排序的原理是将待排序序列分为已排序区间和未排序区间,每次从未排序区间中取出一个元素插入到已排序区间的正确位置。重复这个过程,直到整个序列都变成有序的。

def insertion_sort(arr):
    n = len(arr)
    for i in range(1, n):
        key = arr[i]
        j = i - 1
        while j >= 0 and arr[j] > key:
            arr[j+1] = arr[j]
            j -= 1
        arr[j+1] = key
    return arr

时间复杂度:O(n^2)

快速排序

快速排序是一种分治算法,它的原理是选取一个基准元素,将小于基准元素的元素移到其左侧,大于基准元素的元素移到其右侧。然后对于左右两侧分别递归进行快速排序。

def quick_sort(arr):
    if len(arr) <= 1:
        return arr
    pivot = arr[0]
    left = []
    right = []
    for i in range(1, len(arr)):
        if arr[i] < pivot:
            left.append(arr[i])
        else:
            right.append(arr[i])
    return quick_sort(left) + [pivot] + quick_sort(right)

时间复杂度:O(nlogn)

归并排序

归并排序是一种分治算法,它的原理是将待排序序列分成两个子序列,对于这两个子序列分别进行归并排序,然后将两个有序子序列合并成一个有序序列。

def merge_sort(arr):
    if len(arr) <= 1:
        return arr
    mid = len(arr) // 2
    left = merge_sort(arr[:mid])
    right = merge_sort(arr[mid:])
    return merge(left, right)

def merge(left, right):
    i = 0
    j = 0
    merged = []
    while i < len(left) and j < len(right):
        if left[i] < right[j]:
            merged.append(left[i])
            i += 1
        else:
            merged.append(right[j])
            j += 1
    merged += left[i:]
    merged += right[j:]
    return merged

时间复杂度:O(nlogn)

堆排序

堆排序利用了堆这种数据结构的性质,它的原理是构建一个堆,然后依次将堆顶元素取出放到已排序序列的末尾,重复这个过程,直到整个序列都变成有序的。

def heap_sort(arr):
    n = len(arr)
    for i in range(n, -1, -1):
        heapify(arr, n, i)
    for i in range(n-1, 0, -1):
        arr[0], arr[i] = arr[i], arr[0]
        heapify(arr, i, 0)
    return arr

def heapify(arr, n, i):
    largest = i
    left = 2*i + 1
    right = 2*i + 2
    if left < n and arr[left] > arr[largest]:
        largest = left
    if right < n and arr[right] > arr[largest]:
        largest = right
    if largest != i:
        arr[i], arr[largest] = arr[largest], arr[i]
        heapify(arr, n, largest)

时间复杂度:O(nlogn)

结束语

以上是Python中的一些排序算法,程序员可以根据实际情况选择合适的算法,从而提高程序的效率。需要注意的是,不同的排序算法对于不同的数据集都有其优缺点,应根据实际情况来选择合适的算法。