小鱼刷leetcode---1365. 有多少小于当前数字的数字

45 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第31天,点击查看活动详情

一 描述

1365. 有多少小于当前数字的数字 - 力扣(LeetCode) (leetcode-cn.com)

给你一个数组 nums,对于其中每个元素 nums[i],请你统计数组中比它小的所有数字的数目。

换而言之,对于每个 nums[i] 你必须计算出有效的 j 的数量,其中 j 满足 j != inums[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

二 分析

  1. 拷贝nums到temp,使用temp数组保持nums原有的顺序
  2. 对nums排序(升序)
  3. 创建map记录 nums[i] 对应的 数量(因为是升序,索引值即为小于nums[i]的数量,需要注意的是i>2时,nums[i]与nums[i-1]是否相等)
  4. 遍历temp, 并从countMap中获取小于对应数字的计数和,存储到新的数组中(保持和temp的对应关系)
  5. 返回 countList

三 答案

func smallerNumbersThanCurrent(nums []int) []int {
	temp := make([]int, 0)
	temp = append(temp, nums...)
	sort.Ints(nums)
	countMap := make(map[int]int)
	for i := 0; i < len(nums); i++ {
		if i == 0 {
			countMap[nums[i]] = 0
		} else if nums[i] == nums[i-1] {
			countMap[nums[i]] = countMap[nums[i-1]]
		} else {
            // 因为nums做过排序处理,因此肯定 nums[i]>=num[i-1]
			countMap[nums[i]] = i
		}
	}
	countList := make([]int, len(nums))
	for i, num := range temp {
		countList[i] = countMap[num]
	}
	return countList
}    

参考

Python 计数排序,100分 【全国最菜100分】 - 有多少小于当前数字的数字 - 力扣(LeetCode) (leetcode-cn.com)