给你一个数组 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)
暴力法略去,用计数排序的方法。先用record存储每个数字的个数,再更新为每个小于等于这个数字的个数。
class Solution {
public int[] smallerNumbersThanCurrent(int[] nums) {
int n=nums.length;
int record[] =new int[101];
for(int i=0;i<n;i++){
record[nums[i]]++;
}
for(int i=1;i<101;i++){
record[i]+=record[i-1];
}
int res[]=new int[n];
for(int i=0;i<n;i++){
if(nums[i]==0)
res[i]=0;
else
res[i]=record[nums[i]-1];
}
return res;
}
}
还有一个类似的方法,先排序,然后从后往前遍历。
注意从后往前遍历非常巧妙,可以保证在hash表中表示比nums[i]小的数字有i个
class Solution {
public int[] smallerNumbersThanCurrent(int[] nums) {
int n=nums.length;
int []origin=new int[n];
for(int i=0;i<n;i++){
origin[i]=nums[i];
}
Arrays.sort(nums);
int []hash=new int[101];
for(int i=n-1;i>=0;i--){
hash[nums[i]]=i; //遍历到头,表示比nums[i]小的数字有i个
}
for(int i=0;i<n;i++){
nums[i]=hash[origin[i]];
}
return nums;
}
}