常用演算法模板

清澈的澈發表於2024-11-02

快速排序

def quick_sort(arr):
    if len(arr) <= 1:  # 基本情況:如果陣列為空或只有一個元素,則返回
        return arr
    else:
        pivot = arr[0]  # 選擇基準值(可以選擇第一個元素)
        less_than_pivot = [x for x in arr[1:] if x <= pivot]  # 小於等於基準值的元素
        greater_than_pivot = [x for x in arr[1:] if x > pivot]  # 大於基準值的元素

        # 遞迴排序子列表,併合並結果
        return quick_sort(less_than_pivot) + [pivot] + quick_sort(greater_than_pivot)

歸併排序

def merge_sort(arr):
    # 如果陣列長度小於等於1,直接返回
    if len(arr) <= 1:
        return arr
    
    # 找到陣列的中間點
    mid = len(arr) // 2
    
    # 遞迴分割陣列
    left_half = merge_sort(arr[:mid])
    right_half = merge_sort(arr[mid:])
    
    # 合併已排序的子陣列
    return merge(left_half, right_half)

def merge(left, right):
    sorted_array = []
    i = j = 0
    
    # 合併兩個子陣列
    while i < len(left) and j < len(right):
        if left[i] < right[j]:
            sorted_array.append(left[i])
            i += 1
        else:
            sorted_array.append(right[j])
            j += 1
    
    # 追加剩餘元素
    sorted_array.extend(left[i:])
    sorted_array.extend(right[j:])
    
    return sorted_array

相關文章