Divide and Conquer~ 计算inversions

最新推荐文章于 2019-09-10 00:06:03 发布

原创最新推荐文章于 2019-09-10 00:06:03 发布 · 478 阅读

0 ·

CC 4.0 BY-SA版权

Algorithms 同时被 2 个专栏收录

13 篇文章

订阅专栏

python

12 篇文章

订阅专栏

本文介绍了一种基于归并排序算法来计算数组中逆序对数量的方法。通过对标准归并排序进行简单修改，在排序的同时计算出逆序对的数量。用户可以通过输入文件名来读取数据，然后运行程序得到逆序对的具体数目。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

The code is built on piggybacking on Merge Sort:

x = 0

def mergesort(arr):
    global x
    if len(arr) == 1:
        return arr
    else:
        mid = len(arr) / 2
        left  = mergesort(arr[:mid])
        right = mergesort(arr[mid:])

        left_cnt, right_cnt, arr_cnt = 0, 0, 0

        while left_cnt < len(left) and right_cnt < len(right):
            if left[left_cnt] < right[right_cnt]:
                arr[arr_cnt] = left[left_cnt]
                left_cnt += 1
            else:
                arr[arr_cnt] = right[right_cnt]
                right_cnt += 1; x += (len(left) - left_cnt) ##this is the only difference~!
                
            arr_cnt += 1

        while left_cnt < len(left):
            arr[arr_cnt] = left[left_cnt]
            left_cnt += 1; arr_cnt += 1

        while right_cnt < len(right):
            arr[arr_cnt] = right[right_cnt]
            right_cnt += 1; arr_cnt += 1; 
            
        return arr
    
def main():

    arr = []
    fname = raw_input('please enter the filename:')
    fhand = open(fname)

    for line in fhand:
        line.rstrip()
        arr.append(int(line))
    mergesort(arr)
    print 'x is the number of inversions in the array: '
    print x


main()