题目描述:
给你一个数组 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)。
思路:
首先记录每个数字出现的次数
然后统计每个数字比它小的数字个数
最后按照nums输出
代码如下:
C++:
class Solution {
public:
vector<int> smallerNumbersThanCurrent(vector<int>& nums) {
vector<int>res;
map<int,int>arr;
for(int i:nums){
arr[i]++;
}
int cnt=0;
for(int i=0;i<=100;i++){
int temp=arr[i];
arr[i]=cnt;
cnt+=temp;
}
for(int num:nums){
res.push_back(arr[num]);
}
return res;
}
};
python:
class Solution:
def smallerNumbersThanCurrent(self, nums: List[int]) -> List[int]:
cnt=[0]*101
for num in nums:
cnt[num]+=1
for i in range(1,101):
cnt[i]+=cnt[i-1]
return [0 if num==0 else cnt[num-1] for num in nums]