小朋友排队（归并排序， python）

思路：
  题目要求可以转化为求每个小朋友前后的逆序数目，然后对每个小朋友的逆序数目使用高斯公式，最后累加得到结果。
  求逆序数对可以采用树状数组，本例借鉴了小朋友排队题解-归并排序，更加简洁有效。在归并排序的merge阶段，下标的变化就是存在的逆序对数量，只需要每次累加下标变化即可。
  其实最终的思想还是分治，只是过程刚好和归并排序相同。

# 思路：题目要求小朋友的身高在逆序对中出现的次数，思路是：
# 在归并排序中下标移动的数量就是变化数
# 通过归并排序来计算逆序对数量（其实就是分治的思想）
n = int(input())
children = [int(x) for x in input().split()]

# 三个参数分别是[身高, 原下标, 累计下标变化]，temp是merge时用的临时数组。
children = [[children[i], i, 0] for i in range(n)]
temp = [[0, i, 0] for i in range(n)]

# 归并
def merge(l, mid, r):
    global children, temp
    i = l
    k = l
    j = mid +1
    while i<=mid and j<=r:
        if children[i]<=children[j]:
            temp[k] = children[i][:]
            i+=1
            k+=1
        else:
            temp[k] = children[j][:]
            j+=1
            k+=1
    while i<=mid:
        temp[k]=children[i][:]
        i+=1
        k+=1
    while j<=r:
        temp[k] = children[j][:]
        j+=1
        k+=1
	# 前面都和归并排序相同，只是这里要复制回原数组并累加下标变化
    for k in range(l, r+1):
        children[k] = [temp[k][0], k, temp[k][2]+abs(temp[k][1]-k)]

def mergesort(l, r):
    global children
    if l>= r:
        return
    else:
        mid = (l+r)//2
        mergesort(l, mid)
        mergesort(mid+1, r)
        merge(l, mid, r)

mergesort(0, n-1)
res = 0
for x in children:
    res += x[2]*(x[2]+1)//2

print(res)