题目要求:
给你一个数组 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]
提示:
2 <= nums.length <= 500
0 <= nums[i] <= 100
我的代码:
我的思路: 先将数组A由小到大排序,保存为数组B; 根据原先的数组,找到B中元素的位置,得到比其小的元素的个数。
def smallerNumbersThanCurrent1(nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
sortedList = sorted(nums) #排序后的数组
# print(sortedList)
result = [] # 保存结果的数组
for i in range(0,len(nums)):
for j in range(0,len(sortedList)):
if nums[i] == sortedList[j]:
result.append(j)
break
return result
第一遍做完实现了功能,本来以为提交会超时,没想到通过了,就是效率很低。
所以自己想着优化这个查找算法,→失败了。
最后看了一下官方的题解,效率比较高。
#官方题解:计数排序
def smallerNumbersThanCurrent(nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
nn = [0]*101 #建立空数组,用来存储遍历后的结果,因为nums里的元素大于0小于100(注意这里是101个0.)
# print("nn:",nn)
result = [] # 保存结果的数组
for i in nums:
nn[i] += 1 #遍历到一个数字,nn的对应位置+1
# print("nn:",nn)
for i in nums:
result.append(sum(nn[0:i]))
return result
(在写代码的过程中出现过这个混乱,特此注明。)
break跳出本层循环体,continue跳过本次循环中余下尚未执行的语句,仅结束本次循环。