给你一个数组 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
,取值范围窄,常见的利用桶排序可以简化的题目类型之一。
count
,统计1~100
不同位置上分别存在多少个在nums
数组中对应的值。sum
,sum[k]
的值代表有sum[k]
个nums
数组中的值属于0~k
范围内。nums[i]
,就是统计0~num[k]
存在多少个nums
数组中的值并减去等于nums[i]
的值,即count[nums[i]]
。代码(C语言)
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#define MAXN 1010
int* smallerNumbersThanCurrent(int* nums, int numsSize, int* returnSize){
const int maxn = 110;
int count[maxn],sum[maxn],i=0;
memset(count,0,sizeof(int)*maxn);
memset(sum,0,sizeof(int)*maxn);
for(i=0;i<numsSize;i++)
{
count[nums[i]]++;
}
sum[0] = count[0];
for(i =1;i<maxn;i++)
{
sum[i] = sum[i-1] + count[i];
}
*returnSize = numsSize;
int *res = (int *)malloc(sizeof(int)*numsSize);
for(i=0;i<numsSize;i++)
{
res[i] = sum[nums[i]]-count[nums[i]];
}
return res;
}
int main()
{
int a[] = {5,0,10,0,10,6};
int rsize = 0;
int *b = smallerNumbersThanCurrent(a,6,&rsize);
int i = 0;
for(i=0;i<rsize;i++)
{
printf("%d%c",b[i],i==rsize-1?'\n':' ');
}
}