315. 计算右侧小于当前元素的个数

315. 计算右侧小于当前元素的个数

原始题目链接：https://leetcode-cn.com/problems/count-of-smaller-numbers-after-self/
给你一个整数数组 nums ，按要求返回一个新数组 counts 。数组 counts 有该性质： counts[i] 的值是 nums[i] 右侧小于 nums[i] 的元素的数量。

示例 1：

输入：nums = [5,2,6,1]
输出：[2,1,1,0]
解释：
5 的右侧有 2 个更小的元素 (2 和 1)
2 的右侧仅有 1 个更小的元素 (1)
6 的右侧有 1 个更小的元素 (1)
1 的右侧有 0 个更小的元素
示例 2：

输入：nums = [-1]
输出：[0]
示例 3：

输入：nums = [-1,-1]
输出：[0,0]

解题思路：

使用了内置的二分查找函数，简单的想法是两层for循环，但时间复杂度不满足要求，所以在第二层循环这块需要优化，如果每次比较的时候右侧的元素是有序的，那么就能很快获得元素的个数，bisect.bisect_left(a,x)函数是在 a 中找到 x 合适的插入点以维持有序如果 x 已经在 a里存在，那么插入点会在已存在元素之前（也就是左边），具体实现看代码。

代码实现：

class Solution:
    def countSmaller(self, nums: List[int]) -> List[int]:
        n = len(nums)
        # 赋初值是为了方便插入修改值
        res = [0] * n 
        if not nums:
            return res
        
        # 存储nums从最右边开始，排好序的数组
        sort = []
        # 倒序遍历，为了将nums的右边的数组排序，降低时间复杂度
        for i in range(n - 1, -1, -1):
            # 使用二分查找，找到nums[i]在sort中的位置
            # bisect.bisect_left会返回nums[i]在sort数组中第一个
            # 比sort中所有元素大的位置，相等的元素返回左边第一个索引处
            # 返回的索引就是比较后的大小的个数
            index = bisect.bisect_left(sort, nums[i])
            # 找到位置后，将nums[i]插入到sort数组中合适的位置
            sort.insert(index, nums[i])
            # 对应的i插入对应的大小
            res[i] = index
        
        return res

参考文献：
https://leetcode-cn.com/problems/count-of-smaller-numbers-after-self/solution/jie-jin-shuang-100-jie-ti-fang-an-by-jamleon/