Leetcode 315. Count of Smaller Numbers After Self

这道题采用的是merge sort的思想。

假设已知数组左半部分和右半部分Count of Smaller Numbers After Self结果以及排序结果，还需要知道merge过程中左半部分与右半部分inversions pairs。

在merge过程中，假设i为左半部分数组a的指针，j为右半部分数组b的指针。

1.在任意一步中，如果b[j] <= a[i]，则将b[j]放入有序数组中，并将j指针加1。

2.否则a[i] < b[j]，将a[i]放入有序数组中，并将i指针加1，此时，a[i]有j-1个数比它小且位于它的右边，更新Count of Smaller Numbers After Self结果。

class Solution(object):
    def subCountSmaller(self, start_idx, nums):
        if not nums:
            return [], []
        if len(nums) == 1:
            return [0], nums
        
        mid = len(nums) // 2
        left_count, left_nums = self.subCountSmaller(start_idx, nums[:mid])
        right_count, right_nums = self.subCountSmaller(start_idx + mid, nums[mid:])
        new_nums = []
        left_idx, right_idx = 0, 0
        while len(new_nums) < len(nums):
            if left_idx >= len(left_nums):
                new_nums.extend(right_nums[right_idx:])
                break
            if right_idx >= len(right_nums):
                for i in range(left_idx, len(left_nums)):
                    new_nums.append(left_nums[i])
                    left_count[left_nums[i][1] - start_idx] += len(right_nums)
                break
            if left_nums[left_idx][0] <= right_nums[right_idx][0]:
                new_nums.append(left_nums[left_idx])
                left_count[left_nums[left_idx][1] - start_idx] += right_idx
                left_idx += 1
            else:
                new_nums.append(right_nums[right_idx])
                right_idx += 1
        return left_count + right_count, new_nums
        
    
    def countSmaller(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        nums_idx = []
        for i, num in enumerate(nums):
            nums_idx.append([num, i])
        result, sorted_nums =  self.subCountSmaller(0, nums_idx)
        return result