Как рассчитать инверсию с помощью сортировки слиянием? - PullRequest
0 голосов
/ 11 июня 2018

Я пытаюсь подсчитать общее время инверсии, используя 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. Я думаю, что я, вероятно, делаю какую-то ошибку в своем коде, но я не знаю, как это исправить ... Был бы признателен, если бы кто-то мог мне помочь ...

Ответы [ 3 ]

0 голосов
/ 11 июня 2018
# 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 строки в вашем коде

0 голосов
/ 11 июня 2018

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

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])
0 голосов
/ 11 июня 2018

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

...