LeetCode 1365 有多少小于当前数字的数字 HERODING的LeetCode之路

给你一个数组 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

解题思路:
看到提示相信大家也都知道用什么方法了,范围只有100,那不如用桶排序,首先用桶排序计算每个出现数字的数量,然后遍历桶,统计到某个位置,那么该位置的小于当前数字的数字就是统计到该位置的数字,代码如下:

class Solution {
    
    
public:
    vector<int> smallerNumbersThanCurrent(vector<int>& nums) {
    
    
        int arr[101];
        memset(arr, 0, sizeof(arr));
        // 初始化计数桶
        for (auto i : nums) {
    
    
            arr[i] ++;
        }
        // 累加处理计数桶,使得 arr[i] 表示比 i 小的数字的个数
        int cnt = 0;
        for (int i = 0; i <= 100; i ++) {
    
    
            int temp = arr[i];
            arr[i] = cnt;
            cnt += temp;
        }
        vector<int> ret;
        // 遍历 nums,取出对应桶 arr[i] 里的结果即可
        for (int i : nums) {
    
    
            ret.push_back(arr[i]);
        }
        return ret;
    }
};

猜你喜欢

转载自blog.csdn.net/HERODING23/article/details/109282473
今日推荐