题目
给你一个数组 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]
C++
lower_bound()函数在first和last中的前闭后开区间进行二分查找,返回大于或等于val的第一个元素位置;如果所有元素都小于val,则返回last的位置.
- 先对原数组nums排序
- 通过lower_bound()函数可以获得第一个比num[i]大的数的位置,由于数组是有序的,因此将该下标与首位置相减即可获得比nums[i]小的数的数目,将其保存下来即可
class Solution {
public:
vector<int> smallerNumbersThanCurrent(vector<int>& nums) {
vector<int> temp = nums;
sort(temp.begin(), temp.end());
vector<int> res;
for(int i=0; i<nums.size(); i++){
//lower_bound()函数在first和last中的前闭后开区间进行二分查找,返回大于或等于val的第一个元素位置;
//如果所有元素都小于val,则返回last的位置.
int index = lower_bound(temp.begin(), temp.end(), nums[i]) - temp.begin();
res.push_back(index);
}
return res;
}
};
Python
python可以采用字典进行记录, 用keys记录nums数组中的元素,value是比当前key值小的元素的数目;
采用双重循环,对于每一个nums[i],都将其与所有的key相比较
dict.keys()可以获得一个字典的所有key组成的列表
class Solution:
def smallerNumbersThanCurrent(self, nums: List[int]) -> List[int]:
dic = {}
for i in nums:
dic[i] = 0
for i in nums:
for key in dic.keys():
if key > i:
dic[key] += 1
return [dic[i] for i in nums]