leetcode(py)12

1365. 有多少小于当前数字的数字

难度简单52

给你一个数组 nums，对于其中每个元素 nums[i]，请你统计数组中比它小的所有数字的数目。

换而言之，对于每个 nums[i] 你必须计算出有效的 j 的数量，其中 j 满足 j != i 且 nums[j] < nums[i] 。

以数组形式返回答案。

 

示例 1：

输入：nums = [8,1,2,2,3]
输出：[4,0,1,1,3]
解释： 
对于 nums[0]=8 存在四个比它小的数字：（1，2，2 和 3）。 
对于 nums[1]=1 不存在比它小的数字。
对于 nums[2]=2 存在一个比它小的数字：（1）。 
对于 nums[3]=2 存在一个比它小的数字：（1）。 
对于 nums[4]=3 存在三个比它小的数字：（1，2 和 2）。

我的：

class Solution:
    def smallerNumbersThanCurrent(self, nums: List[int]) -> List[int]:
        return [len([x for x in nums if x<nums[i]]) for i in range(len(nums))]

参考1：（暴力法）

class Solution:
    def smallerNumbersThanCurrent(self, nums: List[int]) -> List[int]:
        n = len(nums)
        vec = [0] * n
        for i in range(n):
            vec[i] = sum(1 for j in range(n) if nums[j] < nums[i])
        return vec

参考2：（频次数组 + 前缀和）(计数排序）（计数排序是一种特殊的桶排序，一般适用于排序数据长度n远大于种类k的情况。比如本题k=101,n=500,甚至5000。）

class Solution:
    def smallerNumbersThanCurrent(self, nums: List[int]) -> List[int]:
        n = len(nums)
        cnt, vec = [0] * 8, [0] * n # nums[8,1,2,2,3]
        for num in nums:
            cnt[num] += 1 # [0,1,2,1,0,0,0,0,1]
        for i in range(1, 9): # 从1开始
            cnt[i] += cnt[i - 1] # [0,1,3,4,4,4,4,4,5]
        for i in range(n):
            if nums[i]:
                vec[i] = cnt[nums[i] - 1] # [4,0,1,1,3]
                # vec[cnt[nums[i]-1]] = nums[i]
                # cnt[nums[i]-1] += 1
        return vec

 

计数排序 

举个例子， nums=[2, 1, 3, 1, 5] , 首先扫描一遍获取最小值和最大值， maxValue=5 , minValue=1 ，于是开一个长度为5的计数器数组 counter ，
1. 分配。统计每个元素出现的频率，得到 counter=[2, 1, 1, 0, 1] ，例如 counter[0] 表示值 0+minValue=1 出现了2次。
2. 收集。 counter[0]=2 表示 1 出现了两次，那就向原始数组写入两个1， counter[1]=1 表示 2 出现了1次，那就向原始数组写入一个2，依次类推，最终原始数组变为 [1,1,2,3,5] ，排序好了。

计数排序本质上是一种特殊的桶排序，当桶的个数最大的时候，就是计数排序。 

不管是vec[i] = cnt[nums[i] - 1] # [4,0,1,1,3]还是vec[i] = cnt[nums[i]]#[5,1,3,3,4]，都存在重复位置，但是排序需要unique值，因此加上 

nums = [8,1,2,2,3]
cnt = [0,1,3,4,4,4,4,4,5]
vec[cnt[nums[i]]-1] = nums[i]
cnt[nums[i]] -= 1

vec[cnt[nums[i]-1]] = nums[i]
cnt[nums[i]-1] += 1

当i=2或者3时，nums[i]是相同的，都为2，因此cnt[nums[i]]-1始终是一个相同的值2，因此vec的值取不到3，结果的3就是空的（0）

当加入下面语句，i=2时，将cnt[2]的值减一，将[5,1,3,3,4]的第二个3变成2，同理，另一种方法是将[4,0,1,1,3]的第二个1变成2。 

参考3：（排序）

class Solution:
    def smallerNumbersThanCurrent(self, nums: List[int]) -> List[int]:
        n = len(nums)
        vec = [0] * n
        tmp = sorted([(nums[i], i) for i in range(n)])
        # [(1, 1), (2, 2), (2, 3), (3, 4), (8, 0)]前面为数字，后面为在nums中的位置，列中的位置为大小排序位置
        
        pre = -1
        for i in range(n):
            if i != 0 and tmp[i][0] != tmp[i - 1][0]:
                pre = i - 1
            vec[tmp[i][1]] = pre + 1
        return vec
# nums=[8,1,2,2,3]
# i=0计算的是最小的，比最小的还小的个数为0
# 正常的非第一个，非重复数，比最小的还小的个数就是i
# tmp[i][1]的意义是与nums顺序对应
# tmp[i][0] != tmp[i - 1][0]计算的是重复数，比如i=2，跳过if后等于还是上次的pre，然后还是上次的i值

排序算法如下 

sorted([nums[i] for i in range(n)])