思路:
题目要求可以转化为求每个小朋友前后的逆序数目,然后对每个小朋友的逆序数目使用高斯公式,最后累加得到结果。
求逆序数对可以采用树状数组,本例借鉴了小朋友排队题解-归并排序,更加简洁有效。在归并排序的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)