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)])