给你一个数组 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
方法一:暴力
比较容易想到的一个方法是,对于数组中的每一个元素,我们都遍历数组一次,统计小于当前元素的数的数目。
class Solution {
public:
vector<int> smallerNumbersThanCurrent(vector<int>& nums) {
vector<int> res(nums.size(), 0);
for(int i=0; i<nums.size(); i++){
int tmp = 0;
for(int j=0; j< nums.size(); j++)
if (nums[j] < nums[i] && i!=j)
tmp += 1;
res[i] = tmp;
}
return res;
}
};
复杂度分析
- 时间复杂度:O(N^2)O(N2),其中 NN 为数组的长度。
- 空间复杂度:O(1)O(1)。注意我们不计算答案数组的空间占用。
方法二:计数排序
注意到数组元素的值域为 [0,100],所以可以考虑建立一个频次数组 cnt,cnt[i]表示数字 ii出现的次数。那么对于数字 ii而言,小于它的数目就为 cnt[0...i-1]的总和。
class Solution {
public:
vector<int> smallerNumbersThanCurrent(vector<int>& nums) {
vector<int> tmp(101, 0);
for(int i=0; i<nums.size(); i++)
tmp[nums[i]]++;
for(int j=1; j<tmp.size(); j++)
tmp[j] += tmp[j-1];
vector<int> res(nums.size(), 0);
for(int k=0; k<nums.size(); k++){
res[k] = nums[k] == 0 ? 0 : tmp[nums[k]-1];
}
return res;
}
};
复杂度分析
时间复杂度:O(N + K),其中 K 为值域大小。需要遍历两次原数组,同时遍历一次频次数组 cnt找出前缀和。
空间复杂度:O(K)。因为要额外开辟一个值域大小的数组。