python合并排序问题

时间:2016-01-26 12:29:48

标签: python algorithm sorting mergesort clrs

不确定我在python中实现合并排序的方法出错了。

import sys

sequence = [6, 5, 4, 3, 2, 1]

def merge_sort(A, first, last):
    if first < last:
        middle = (first + last) / 2
        merge_sort(A, first, middle)
        merge_sort(A, middle+1, last)
        merge(A, first, middle, last)

def merge(A, first, middle, last):
    L = A[first:middle]
    R = A[middle:last]

    L.append(sys.maxint)
    R.append(sys.maxint)

    i = 0
    j = 0
    for k in xrange(first, last):
        if L[i] <= R[j]:
            A[k] = L[i]
            i = i + 1
        else:
            A[k] = R[j]
            j = j + 1

merge_sort(sequence, 0, len(sequence))
print sequence

如果有人能够指出什么打破了我目前的合并排序实现,我真的很感激。

2 个答案:

答案 0 :(得分:5)

代码中有2个错误:

  • if first < last:应为if first < last and last-first >= 2:
  • merge_sort(A, middle+1, last)应为merge_sort(A, middle, last)

答案 1 :(得分:4)

问题在于:

    merge_sort(A, first, middle)
    merge_sort(A, middle+1, last) # BEEP

你只能从中间+ 1排序第二部分,你应该从中间开始。事实上,你永远不会在中间重新排序元素。

当然你不能写

    merge_sort(A, first, middle)
    merge_sort(A, middle, last) # BEEP, BEEP

因为当last = first + 1时,你得到中间==然后潜入无限递归(由RuntimeError: maximum recursion depth exceeded停止)

所以要走的路是:

    merge_sort(A, first, middle)
    if middle > first: merge_sort(A, middle, last)

经过一点改变后,您的实施会得到正确的结果。