题目描述
给你一个数组 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
来源:力扣(LeetCode)
题解
作为一个简单题,第一眼上去,啊这不是我熟悉的暴力解法吗?循环就完了!
//Brute Force
class Solution {
public:
vector<int> smallerNumbersThanCurrent(vector<int>& nums) {
vector<int> res;
for(int i=0;i<nums.size();i++){
res.push_back(0);
for(int j=0;j<nums.size();j++){
if(i==j) continue;
if(nums[i]>nums[j]) res[i]++;
}
}
return res;
}
};
但是!8行!今天的任务不能就这样糊弄过去了,不求甚解,今天就等于白给。继续优化!
既然是找比自己小的数,那肯定存在重复计算,而一般这种问题都是空间换取时间。
我们用一个数组来存每个数字的出现频率,nums数组最大为100,因此一个大小为101的数组就够用了,遍历一遍,计算每个数字的出现频率。通过累加,将数组记录为小于当前数字的数字个数。
class Solution{
public:
vector<int> smallerNumbersThanCurrent(vector<int>& nums){
vector<int> res;
int freq[101]{};
for(int i=0;i<nums.size();i++){ //记录每个数字的出现频率
freq[nums[i]]++;
}
int cur=0;
for(int i=0;i<101;i++){ //将数组每一位累加为小于当前数字的数字个数
int tem = freq[i];
freq[i] = cur;
cur+=tem;
}
for(int i:nums){ //直接读结果就完了
res.push_back(freq[i]);
}
return res;
}
};