参考链接
- https://leetcode-cn.com/problems/how-many-numbers-are-smaller-than-the-current-number/
- https://leetcode-cn.com/problems/how-many-numbers-are-smaller-than-the-current-number/solution/you-duo-shao-xiao-yu-dang-qian-shu-zi-de-shu-zi–2/
题目描述
给你一个数组 nums,对于其中每个元素 nums[i],请你统计数组中比它小的所有数字的数目。
换而言之,对于每个 nums[i] 你必须计算出有效的 j 的数量,其中 j 满足 j != i 且 nums[j] < nums[i] 。以数组形式返回答案。
解题思路
暴力解法自不必多说。本题最好是先排序,排序后的数组中,每个元素左边的元素就是比它小的。接下来就是考虑一些细节问题。值得注意的是,本题将相同值的元素视为不同的元素,如比8小的数有1、2、2、3,而不是1、2、3。我的思路是,如果当前元素与前一个元素值相同,那么比它们小的元素数量一样。否则,比当前元素小的元素数量就是比前一个元素小的数量加1,但是要考虑前一个元素重复的情况。所以引入了一个变量bias,用来记录前一个元素重复出现的次数,最后需要再加上bias。
官方的代码写的很简洁,一是将原数组元素和索引存在一个vector里,而我却多复制了一份nums;二是其只使用了一次遍历,而我却使用了两次遍历,一次将元素和比它小的元素个数存储在哈希表里,二次利用复制的数组从哈希表里取值产生结果。实在是菜,罪过罪过。
代码
我的代码
class Solution {
public:
vector<int> smallerNumbersThanCurrent(vector<int>& nums) {
vector<int> copy_nums(nums);
vector<int> res(nums.size());
int bias = 0;
unordered_map<int, int> mp;
sort(nums.begin(), nums.end());
mp[nums[0]] = 0;
for (int i = 1; i < nums.size(); i ++)
{
if (nums[i] == nums[i - 1])
{
mp[nums[i]] = mp[nums[i - 1]];
bias ++;
}
else
{
mp[nums[i]] = mp[nums[i - 1]] + 1 + bias;
bias = 0;
}
}
for (int i = 0; i < nums.size(); i ++)
{
res[i] = mp[copy_nums[i]];
}
return res;
}
};
官方代码
class Solution {
public:
vector<int> smallerNumbersThanCurrent(vector<int>& nums) {
vector<pair<int, int>> data;
int n = nums.size();
for (int i = 0; i < n; i++) {
data.emplace_back(nums[i], i);
}
sort(data.begin(), data.end());
vector<int> ret(n, 0);
int prev = -1;
for (int i = 0; i < n; i++) {
if (prev == -1 || data[i].first != data[i - 1].first) {
prev = i;
}
ret[data[i].second] = prev;
}
return ret;
}
};