leetcode专题训练315. Count of Smaller Numbers After Self

一开始就联想到了用归并排序求逆序数这道题，可是当时求逆序数的时候，不需要知道逆序数出现在哪个位置，只用记录逆序数个数即可，所以改了一下当时的代码，利用index数组记录某位置的元素的初始位置，再通过result[index[pos]] += 1即可把当前逆序数加入到原位置的结果。然而我写的加入到原位置的代码会超时。
超时代码：

class Solution:
    

    def countSmaller(self, nums: List[int]) -> List[int]:
        l = len(nums)
        if l == 0:
            return []

        result = [0 for i in range(l)]
        # 需要用此得到某位置元素最后挪到的位置
        # 对应关系：结果位置->初始位置
        index = [i for i in range(l)]
        # 归并排序
        def Sort(left, right):
            if left < right:
                mid = (left+right) >> 1
                # print(left, mid, right)
                Sort(left, mid)
                Sort(mid+1, right)
                Merge(left, mid, right)

        def Merge(left, mid, right):
            tmp = []
            tmp_index = [0 for i in range(right-left+1)]
            i = left
            j = mid+1
            k = 0
            while i <= mid and j <= right:
                if nums[i] > nums[j]:
                    tmp.append(nums[j])
                    # 逆序数更新，只要出现一个逆序数，那么就更新
                    for pos in range(i, mid+1):
                        result[index[pos]] += 1
                    tmp_index[k] = index[j]
                    j += 1
                else:
                    tmp.append(nums[i])
                    tmp_index[k] = index[i]
                    i += 1
                k += 1
            while i <= mid:
                tmp.append(nums[i])
                tmp_index[k] = index[i]
                k += 1
                i += 1
            while j <= right:
                tmp.append(nums[j])
                tmp_index[k] = index[j]
                k += 1
                j += 1
            nums[left:right+1] = tmp[:]
            index[left:right+1] = tmp_index[:]

        Sort(0, l-1)
        return result

后来改了一下逆序数的更新方式。将某个位置的逆序数积攒在一起更新，而不是有一个该位置逆序数就更新，改进方式是将更新逆序数放在下面代码中的位置，具体看注释。需要注意的是增加的逆序数个数需要自己模拟一下程序过程，推一下。
可通过代码：

class Solution:
    

    def countSmaller(self, nums: List[int]) -> List[int]:
        l = len(nums)
        if l == 0:
            return []

        result = [0 for i in range(l)]
        # 需要用此得到某位置元素最后挪到的位置
        # 对应关系：结果位置->初始位置
        index = [i for i in range(l)]
        # 归并排序
        def Sort(left, right):
            if left < right:
                mid = (left+right) >> 1
                # print(left, mid, right)
                Sort(left, mid)
                Sort(mid+1, right)
                Merge(left, mid, right)

        def Merge(left, mid, right):
            tmp = []
            tmp_index = [0 for i in range(right-left+1)]
            i = left
            j = mid+1
            k = 0
            while i <= mid and j <= right:
                if nums[i] > nums[j]:
                    tmp.append(nums[j])
                    tmp_index[k] = index[j]
                    j += 1
                else:
                    # 找到该位置此次排序的所有逆序数，更新
                    result[index[i]] += j-mid-1
                    tmp.append(nums[i])
                    tmp_index[k] = index[i]
                    i += 1
                k += 1
            while i <= mid:
                tmp.append(nums[i])
                tmp_index[k] = index[i]
                # 找到该位置此次排序的所有逆序数，更新
                result[index[i]] += (right-mid)
                k += 1
                i += 1
            while j <= right:
                tmp.append(nums[j])
                tmp_index[k] = index[j]
                k += 1
                j += 1
            nums[left:right+1] = tmp[:]
            index[left:right+1] = tmp_index[:]

        Sort(0, l-1)
        return result