题目
给你一个数组 nums,对于其中每个元素 nums[i],请你统计数组中比它小的所有数字的数目。
换而言之,对于每个 nums[i] 你必须计算出有效的 j 的数量,其中 j 满足 j != i 且 nums[j] < nums[i] 。
以数组形式返回答案。
提示:
2 <= nums.length <= 500
0 <= nums[i] <= 100
题目地址:https://leetcode-cn.com/problems/how-many-numbers-are-smaller-than-the-current-number/
思路
核心思想:将nums数组的值当作cnt数组的索引
代码
class Solution {
public:
vector<int> smallerNumbersThanCurrent(vector<int>& nums) {
vector<int> cnt(101, 0); //创建一个大小为101,初始值为0的数组
vector<int> vec((int)nums.size(), 0);
for (int i = 0;i < (int)nums.size(); ++i){
cnt[nums[i]]++;
}
for (int i = 1;i <= 100; ++i) cnt[i] += cnt[i-1]; // 求前缀和
for (int i = 0;i < (int)nums.size(); ++i){
if (nums[i]) vec[i] = cnt[nums[i] - 1];
}
return vec;
}
};