1365 有多少小于当前数字的数字
题目 1365 有多少小于当前数字的数字
给你一个数组 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)。
示例 2:
输入:nums = [6,5,4,8]
输出:[2,1,0,3]
思路
- 发现如果数组从小到大排序,那么数组元素的下标是不是就是小于该元素的数字个数,所以可以用哈希字典来存储,key是数字,value是该数字对应小于数字的数量结果。
- 如果有重复元素怎么办呢?可以对有序数组倒序遍历即可。
- 最后循环原数组,从字典中找到对应的结果即可。注意:排序后原数组本来的值发生变换,如果一开始init_nums = nums保留一份,也不可以,该赋值意味着init_nums变量指向nums变量而已,nums变量发生变化,init_nums也发生变化,具体知识可以去了解python的引用知识。所以可以利用这个技巧init_nums = nums[:]即可。
代码
class Solution:
def smallerNumbersThanCurrent(self, nums: List[int]) -> List[int]:
res = [] # 最后结果
res_dict = {} # 中间字典,保存数字和数字下标
init_nums = nums[:] # 保留原数组一份
nums.sort()
nums_len = len(nums)
while nums_len > 0: # 倒序遍历排序后的数组
res_dict[nums[nums_len-1]] = nums_len-1
nums_len -= 1
for i in init_nums:
res.append(res_dict[i])
return res