洛谷P1908 逆序对

题目描述

猫猫 TOM 和小老鼠 JERRY 最近又较量上了，但是毕竟都是成年人，他们已经不喜欢再玩那种你追我赶的游戏，现在他们喜欢玩统计。

最近，TOM 老猫查阅到一个人类称之为“逆序对”的东西，这东西是这样定义的：对于给定的一段正整数序列，逆序对就是序列中ai​>aj​ 且 i<j 的有序对。知道这概念后，他们就比赛谁先算出给定的一段正整数序列中逆序对的数目。注意序列中可能有重复数字。

Update:数据已加强。

输入格式

第一行，一个数 n，表示序列中有 n个数。

第二行 n 个数，表示给定的序列。序列中每个数字不超过 10^9。

输出格式

输出序列中逆序对的数目。

输入输出样例

输入 #1

6
5 4 2 6 3 1

输出 #1

11

分治加归并排序

首先要知道对于某两段数据来说，他们内部的顺序并不会改变这两段数据之间的逆序数。比如

1 6 7 5 | 4 5 1 3

将前面的1675顺序随便改动不会影响每一个数与后面4513的逆序数（不考虑1675内部的逆序数，为什么不考虑后面会提到）

那么我可以把这两段数据分别按升序排列，这样找每一个数的逆序数会更方便。

分治就要合并；分治可以使用递归（类似于二分切成一段一段），合并的时候可以将这两组数据按升序重新排列，合成一组数据。

那么要怎么合并？

i 指向前一段数的第一个数的下标，j 指向后一段数的第一个数的下标；当ai小于aj时，把ai放进一个新的列表 r 里，i+=1；当aj小于ai时，把aj放进这个新的列表 r 里，j+=1.这样生成的新列表就是升序排列的前两段数据的合并，最后再把这个新列表 r

按照对应的位置放进原来的列表里。为什么直接放进去，不考虑会不会有其他逆序数？因为递归拿到的被改动过的原列表都是按照升序排列的。

 mid - i + 1 是从 i 开始到前一段数据的最后一个数总共有多少个数，就是 aj 的逆序数

n = int(input())
lst = list(map(int,input().split()))
r = [0]*n
ans = 0
def msort(start,end):
    global ans          # 先看递归再看 if
    if start == end:    # 当一段数据只有两个时，传进来的参数start == mid
        return
    mid = (start + end)//2
    msort(start,mid)    # 前面一段数据求逆序数
    msort(mid+1,end)
    s = start
    m = mid+1
    e = end
    k = start       # 专门给新列表 r 的
    while s <= mid and m <= e:  # 两段数据都没走完
        if lst[s] <= lst[m]:
            r[k] = lst[s]
            s += 1
            k += 1
        else:
            r[k] = lst[m]
            k += 1
            m += 1
            ans += mid-s+1
    while s <= mid:     # 两段数据有一段走完了但是另一段没走完，把剩下的走完
        r[k] = lst[s]
        k += 1
        s += 1
    while m <= e:
        r[k] = lst[m]
        k += 1
        m += 1
    for i in range(start,end+1):    
        lst[i] = r[i]

msort(0,n-1)        # 从第一个数到最后一个数
print(ans)