使用python实现计数反转的错误结果

Question

我正在尝试使用 python 实现带有 mergeSort 的计数版本，这是我的代码：

def merge(inLeft, inRight):
    inversions = 0; output = []
    while 0 < len(inLeft) and 0 < len(inRight):
        if inLeft[0] < inRight[0]:
            output.append(inLeft[0])
            inLeft.remove(inLeft[0])
        else:
            output.append(inRight[0])
            inRight.remove(inRight[0])
            inversions += len(inLeft)

    if len(inLeft) == 0:
        output.append(inRight[0])
    elif len(inRight) == 0:
        output.append(inLeft[0])    
    return output, inversions

def mergeSort(inList):
    length = len(inList)
    if length == 1:
        return inList, 0
    left, s1 = mergeSort(inList[: length//2])
    right, s2 = mergeSort(inList[length//2: ])
    sortedList, s3 = merge(left, right)
    return sortedList, (s1+s2+s3)

我以为当我通过mergeSort([1, 3, 5, 2, 4, 6]) 调用它时会得到([1, 2, 3, 4, 5, 6], 3)，但实际上我得到了([1, 2, 3, 4], 1)，当我检查它时，我发现left 数组总是<built-in function sorted>。

我正在研究分治算法，因此不擅长递归分析问题。问题可能出在哪里？我该如何解决？

Answer 1

if len(inLeft) == 0:
    output.append(inRight[0])
elif len(inRight) == 0:
    output.append(inLeft[0])

这只会将第一个元素添加到输出中。更改为output.extend(inRight) / output.extend(inLeft) 以有效地添加整个数组。这将修复您缺少的元素。

此外，Python 列表的 remove 操作具有 O(N) 复杂性，因此您可能需要考虑使用双端队列 (collections.deque)，它可以有效地从列表的前面删除。