桶排序的应用—— 有多少小于当前数字的数字（leetcode 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

暴力算法一定是可以算出来的，这里就不说了。

题目中有说 0 <= nums[i] <= 100，有了这个条件，那桶计数就再适合不过了。

如果桶排序你忘了，可以看我之前的博文——桶排序

具体到这个题目，就是用一个101个桶，代表从0至100

把 nums 对应放入桶中，每个桶里的数字是相同的，且桶是有序的。

比如想查看比80小的数总共多少个，那就是 0 至 79 号桶里元素的和。

直接看代码

    public int[] smallerNumbersThanCurrent(int[] nums) {
        int[] cnt = new int[101]; // 桶的个数
        int n = nums.length;
        for (int i = 0; i < n; i++) {
            cnt[nums[i]]++; // 统计每个桶中数字的个数
        }
        for (int i = 1; i <= 100; i++) {
            cnt[i] += cnt[i - 1]; // 累加后，79号桶的值，就是小于80的元素个数
        }
        int[] ret = new int[n]; // 结果输出
        for (int i = 0; i < n; i++) {
            ret[i] = nums[i] == 0 ? 0 : cnt[nums[i] - 1];
        }
        return ret;
    }

很典型的桶计数的应用，我当时没想到，是掌握的不到家呀！