Как рассчитать инверсию с помощью сортировки слиянием?

Question

Я пытаюсь подсчитать общее время инверсии, используя def count_inversion

def count_inversion(alist):
count = 0
if len(alist)>1:
    mid = len(alist)//2
    lefthalf = alist[:mid]
    righthalf = alist[mid:]

    a=count_inversion(lefthalf)
    b=count_inversion(righthalf)

    i=0
    j=0
    k=0
    track = 0
    while i < len(lefthalf) and j < len(righthalf):
        if lefthalf[i] < righthalf[j]:
            alist[k]=lefthalf[i]
            i=i+1

        else:
            alist[k]=righthalf[j]
            j=j+1
            count+=len(righthalf[i:])
        k=k+1

    while i < len(lefthalf):
        alist[k]=lefthalf[i]
        i=i+1
        k=k+1


    while j < len(righthalf):
        alist[k]=righthalf[j]
        j=j+1
        k=k+1

return count

def main():
    alist = [10,9,8,7,6,5,4,3,2,1]
    inversion = count_inversion(alist)
    print(alist)
    print(inversion)    

main()

Я получил отсортированный список [1,2,3,4,5,6,7,8,9,10], но для счетчика инверсий он показывает, что это 25 вместо 45. Я думаю, что я, вероятно, допустил некоторую ошибку в мой код, но я понятия не имею, как это исправить... Был бы признателен, если кто-то может мне помочь...

Answer 1

# store inversion count 
count+=count_inversion(lefthalf)
count+=count_inversion(righthalf)

# updated line > Instead of righthalf count length of lefthalf
count+=len(lefthalf[i:])

Обновите эти 2 строки в вашем коде

Answer 2

Вот моя реализация внутри функции класса. Это также считает инверсии. Вы можете раскомментировать операторы, чтобы увидеть альтернативный способ отслеживания инверсий.

class MergeSort(object):
'''
Instantiates an array into the object from which the method 'merge_sort' can be called.
Returns number of inversions and sorted array.
>>>x = MergeSort([1,6,5,2,10,8,7,4,3,9])
>>>x.merge_sort()
(20, [1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
'''
def __init__(self, array):
    self.array = array

def merge_sort(self):
    count = 0
    #count_ = []
    if len(self.array) > 1:
        m = len(self.array)//2
        left = self.array[:m]
        right = self.array[m:]

        leftsorter = MergeSort(left)
        leftsorter = leftsorter.merge_sort()
        rightsorter = MergeSort(right)
        rightsorter = rightsorter.merge_sort()

        # Two different ways to track inversions

        # numeric counter, better way
        count += leftsorter[0]
        count += rightsorter[0]

        # list counter
        #count_.append(leftsorter[0])
        #count_.append(rightsorter[0])

        i = 0
        j = 0
        k = 0

        while i < len(left) and j < len(right):
            if left[i] < right[j]:
                self.array[k] = left[i]
                i += 1
            else:
                self.array[k] = right[j]
                j += 1
                count += len(left[i:])
                #count_.append(len(left[i:]))
            k += 1

        while i < len(left):
            self.array[k] = left[i]
            i += 1
            k += 1

        while j < len(right):
            self.array[k] = right[j]
            j += 1
            k += 1
    return count, self.array, #sum(count_)

Который работает таким образом.

array = [1,6,5,2,10,8,7,4,3,9]
x = MergeSort(array)
x.merge_sort()

Out[ ]:
(20, [1, 2, 3, 4, 5, 6, 7, 8, 9, 10])

Answer 3

Вы не могли бы использовать этот пример кода? https://www.geeksforgeeks.org/counting-inversions/