1365. 有多少小于当前数字的数字

题目

给你一个数组 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）
链接：https://leetcode.cn/problems/how-many-numbers-are-smaller-than-the-current-number
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

做完今天的每日一题1051，顺带复习一下计数排序，之前做过的只找到这一个题。

因为本题的数据范围也很小，所以可以考虑计数排序。

很直接，本题求的小于数组中每个数的元素个数，这正是计数排序的计数数组维护的定义，只不过本题是「小于」而不是「小于等于」。

见代码一。

代码

代码一

class Solution {
    public int[] smallerNumbersThanCurrent(int[] nums) {
        int n = nums.length;
        int[] ans = new int[n];
        // freq[i]表示数字中「小于等于」i的元素个数
        int[] freq = new int[101];
        for (int num : nums) {
            freq[num]++;
        }
        for (int i = 1; i < 101; i++) {
            freq[i] += freq[i - 1];
        }
        for (int i = 0; i < n; i++) {
            // 注意元素可能为0
            ans[i] = nums[i] == 0 ? 0 : freq[nums[i] - 1];
        }
        return ans;
    }
}

代码二

也可以重新定义freq[i]的定义，改为数字中「小于」i的元素个数，这样就不用考虑nums[i]-1溢出的问题了。

class Solution {
    public int[] smallerNumbersThanCurrent(int[] nums) {
        int n = nums.length;
        int[] ans = new int[n];
        // freq[i]表示数字中「小于」i的元素个数
        int[] freq = new int[101];
        for (int num : nums) {
            freq[num]++;
        }
        int sum = 0;
        for (int i = 0; i < 101; i++) {
            int tmp = freq[i];
            freq[i] = sum;
            sum += tmp;
        }
        for (int i = 0; i < n; i++) {
            ans[i] = freq[nums[i]];
        }
        return ans;
    }
}