leetcode 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]
示例 3:
输入:nums = [7,7,7,7]
输出:[0,0,0,0]
步骤:
根据官方题解有三种方法,不过官方题解中并没有Python的代码,所以我自己参考其他人的题解写了一遍。
最容易想到的就是暴力解法了,时间复杂度大概是O(n^2)
class Solution(object):
def smallerNumbersThanCurrent(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
ans=[]
for i in range(len(nums)):
sum=0
for j in range(len(nums)):
if nums[j]<nums[i]:
sum+=1
ans.append(sum)
return ans
第二种是利用排序和哈希表,先将数组排序好,将数组中的元素以及他们的下标(这里是排序好的下标即比他们小的数的个数)存放在字典里面,最后根据nums中的元素顺序分别将他们的下标从字典中取出。
class Solution(object):
def smallerNumbersThanCurrent(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
sortnums=sorted(nums)
dic={}
for i,j in enumerate(sortnums):
if j not in dic:
dic[j]=i
sortnums=[]
for p in nums:
sortnums.append(dic[p])
return sortnums
最后一种方法是计数排序,可以参考这个题解。
计数排序
class Solution(object):
def smallerNumbersThanCurrent(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
list1=[0]*101
for i in nums:
list1[i]+=1
temp=0
list2=[]
for i in list1:
list2.append(temp)
temp+=i
ans=[]
for i in nums:
ans.append(list2[i])
return ans