Question

我很难让这个快速排序功能逐一返回实际列表而不是项目。排序逻辑工作正常，但它没有返回数组。有没有办法添加项目附加到每个循环的持久列表？

import random


def quick_sort_2(input_arr):

    if len(input_arr) == 1:
        print(input_arr[0])
        return input_arr[0]
    elif len(input_arr) == 0:
        return
    else:
        pivot = input_arr[0]
        i = 0
        left_arr = []
        right_arr = []
        while i < len(input_arr):
            if input_arr[i] < pivot:
                left_arr.append(input_arr[i])
                i += 1
            elif input_arr[i] > pivot:
                right_arr.append(input_arr[i])
                i += 1
            else:
                i += 1
        quick_sort_2(left_arr)
        print(pivot)
        quick_sort_2(right_arr)


quick_sort_2([3, 5, 2, 6, 1, 7, 0])

快速注意的是我在这里使用打印，但实际上我想以某种方式使用返回。

Answer 1

解决方案是在函数的每个分支中实际返回结果列表。如果没有要排序的话，它可能是空列表。否则，它是小于pivot元素的所有元素的排序列表，加上pivot元素，以及大于或等于（或不小于）pivot元素的所有元素的排序列表。

def quick_sort_2(items):
    items = iter(items)
    try:
        pivot = next(items)
    except StopIteration:
        return []
    else:
        left_items = []
        right_items = []
        for item in items:
            (left_items if item < pivot else right_items).append(item)
        return quick_sort_2(left_items) + [pivot] + quick_sort_2(right_items)

快速排序附加到列表与递归？

1 个答案: