Question

我正在接受普林斯顿的算法 - 分治 - 征服课程 - 第3周，并尝试实施快速排序。

这是我当前的实现，一些测试准备好运行：

import unittest

def quicksort(x):
    if len(x) <= 1:
        return x

    pivot = x[0]
    xLeft, xRight = partition(x)
    print(xLeft, xRight)
    quicksort(xLeft)
    quicksort(xRight)
    return x


def partition(x):
    j = 0
    print('partition', x)
    for i in range(0, len(x)):
        if x[i] < x[0]:
            n = x[j + 1]
            x[j + 1] = x[i]
            x[i] = n
            j += 1

    p = x[0]
    x[0] = x[j]
    x[j] = p
    return x[:j + 1], x[j + 1:]


class Test(unittest.TestCase):
    def test_partition_pivot_first(self):
        arrays = [
            [3, 1, 2, 5],
            [3, 8, 2, 5, 1, 4, 7, 6],
            [10, 100, 3, 4, 2, 101]
        ]

        expected = [
            [[2, 1, 3], [5]],
            [[1, 2, 3], [5, 8, 4, 7, 6]],
            [[2, 3, 4, 10], [100, 101]]
        ]

        for i in range(0, len(arrays)):
            xLeft, xRight = partition(arrays[i])
            self.assertEqual(xLeft, expected[i][0])
            self.assertEqual(xRight, expected[i][1])

    def test_quicksort(self):
        arrays = [
            [1, 2, 3, 4, 5, 6],
            [3, 5, 6, 10, 2, 4]
        ]

        expected = [
            [1, 2, 3, 4, 5, 6],
            [1, 2, 3, 4, 6, 10]
        ]

        for i in range(0, len(arrays)):
            arr = arrays[i]
            quicksort(arr)
            self.assertEqual(arr, expected[i])


if __name__ == "__main__":
    unittest.main()

因此对于array = [3, 5, 6, 10, 2, 4]我得到了[2, 3, 6, 10, 5, 4] ...我无法弄清楚我的代码有什么问题。它分区很好，但结果都没有了......

任何人都可以进入吗？ :)谢谢！

Answer 1

这实际上是一个小问题，你会笑问题在于快速排序功能正确的是：

def quicksort(x):
 if len(x) <= 1:
    return x

 pivot = x[0]
 xLeft, xRight = partition(x)
 print(xLeft, xRight)
 quicksort(xLeft)
 quicksort(xRight)
 x=xLeft+xRight #this one!
 return x

发生了什么是python从这些xleft和xright中创建了一个新对象，它们从来就不存在 - 排序

所以这是一个解决方案（没有到位）另一个是传递列表，start_index，end_index 并做到位

做得好的家伙！编辑：实际上，如果你打印xleft和xright，你会发现它表现得非常完美：）

Python quicksort只排序上半部分

1 个答案: