改进归并即可
def merge(li, low, mid, high, lis):
i = low
j = mid + 1
ltmp = []
m = 0
while i <= mid and j <= high:
if li[i] <= li[j]:
ltmp.append(li[i])
i += 1
else:
ltmp.append(li[j])
j += 1
m += mid - i + 1
while i <= mid:
ltmp.append(li[i])
i += 1
while j <= high:
ltmp.append(li[j])
j += 1
li[low: high + 1] = ltmp
if m != 0:
lis.append(m)
def mergesort(li, low, high):
if low < high:
mid = (low + high) // 2
mergesort(li, low, mid, )
mergesort(li, mid + 1, high, )
merge(li, low, mid, high, lis)
if __name__ == "__main__":
n = input()
ls = list(map(int, input().split()))
p = 0
lis = []
mergesort(ls, 0, len(ls) - 1)
for i in lis:
p += i
print(p)