为什么这个合并排序实现不起作用?

时间:2016-02-15 06:00:55

标签: python mergesort clrs

我正在关注CLRS书籍,并且正在实现python中的所有算法,这是合并排序算法。我在测试数组上测试了合并函数并且它可以工作,因此它必须是merge_sort函数,但它与本书中的伪代码几乎相同。

def merge_sort(self, array, p, r):
    if p < r:
        q = int((p + r) / 2)

        self.merge_sort(array, p, q)
        self.merge_sort(array, q + 1, r)
        self.merge(array, p, q , r)

def merge(self, array, p, q, r):
    n1 = q - p + 1
    n2 = r - q
    left = []
    right = []

    for i in range(0, n1):
        left.append(array[p + i])

    for j in range (0, n2):
        right.append(array[q + j + 1])

    left.append(math.inf)
    right.append(math.inf)
    i = 0
    j = 0

    for k in range(p, r):
        if left[i] <= right[j]:
            array[k] = left[i]
            i += 1
        else:
            array[k] = right[j]
            j += 1

1 个答案:

答案 0 :(得分:0)

我猜这个:

for k in range(p, r):

应该是:

for k in range(p, r + 1):

这将使k取p到r的所有值,然后将p的值转换为r - 1 - 就像range()的工作方式一样。