使用merge sort python进行反转计数

时间:2014-08-08 18:39:36

标签: python sorting merge count inversion

我正在使用反转计数实现合并排序。 我的代码适用于我的小输入文件(10个元素),但当它变为100000时, 似乎从粗暴的搜索返回错误的答案。有时更大,有时更小。有人有任何想法吗?

我的代码返回2402298631。 输入文件位置 http://spark-public.s3.amazonaws.com/algo1/programming_prob/IntegerArray.txt

def msort_inv2(m):

    global count

    if len(m) <= 1:
        return m
    result = []
    middle = int(len(m)/2)

    left = msort_inv2(m[:middle])
    right = msort_inv2(m[middle:])

    while (len(left) > 0) or (len(right) > 0):
        if (len(left) > 0) and (len(right) > 0):
            if left[0] > right[0]:
                result.append(right[0])
                count = count + len(left)
                right.pop(0)
            else:
                result.append(left[0])
                left.pop(0)
        elif len(right) > 0:
            for i in right:
                result.append(i)
                right.pop(0)
        else:
            for i in left:
                result.append(i)
                left.pop(0)
    return result

1 个答案:

答案 0 :(得分:1)

该错误存在于以下部分

    elif len(right) > 0:
        for i in right:     # bug !!!!
            result.append(i)
            right.pop(0)
    else:
        for i in left:
            result.append(i)
            left.pop(0)

校正将是这样的

    elif len(right) > 0:
        result.extend(right)
        right = []
    else:
        result.extend(left)
        left = []

对于数组中的循环并同时弹出项目,会在python中引起奇怪的行为。

相关问题