算法 力扣 1887. 使数组元素相等的减少操作次数
给你一个整数数组 nums ,你的目标是令 nums 中的所有元素相等。完成一次减少操作需要遵照下面的几个步骤:
- 找出
nums
中的 最大值。记这个值为largest
并取其下标 i (下标从 0 开始计数)。如果有多个元素都是最大值,则取最小的 i 。 - 找出 nums 中的 下一个最大 值,这个值 严格小于
largest
,记为nextLargest
。 - 将
nums[i]
减少到nextLargest
。
返回使 nums 中的所有元素相等的操作次数。
示例 1:
输入:nums = [5,1,3] 输出:3 解释:需要 3 次操作使 nums 中的所有元素相等:
- largest = 5 下标为 0 。nextLargest = 3 。将 nums[0] 减少到 3 。nums = [3,1,3] 。
- largest = 3 下标为 0 。nextLargest = 1 。将 nums[0] 减少到 1 。nums = [1,1,3] 。
- largest = 3 下标为 2 。nextLargest = 1 。将 nums[2] 减少到 1 。nums = [1,1,1] 。
示例 2:
输入:nums = [1,1,1]
输出:0 解释:nums 中的所有元素已经是相等的。
示例 3:
输入:nums = [1,1,2,2,3] 输出:4 解释:需要 4 次操作使 nums 中的所有元素相等:
- largest = 3 下标为 4 。nextLargest = 2 。将 nums[4] 减少到 2 。nums = [1,1,2,2,2] 。
- largest = 2 下标为 2 。nextLargest = 1 。将 nums[2] 减少到 1 。nums = [1,1,1,2,2] 。
- largest = 2 下标为 3 。nextLargest = 1 。将 nums[3] 减少到 1 。nums = [1,1,1,1,2] 。
- largest = 2 下标为 4 。nextLargest = 1 。将 nums[4] 减少到 1 。nums = [1,1,1,1,1] 。
/**
* @param {number[]} nums
* @return {number}
*/
var reductionOperations = function(nums) {
/**
* 所有元素和操作次数,等于比它小的元素的种类数:例如:[5,3,3,1]1和3两种,因此操作为2次
* 将所有元素去重,进行从小到大排序,元素所在位置的索引i就是它的操作次数
*/
// 去重
const removalArr = [...new Set(nums)]
// 排序
removalArr.sort((a,b) => a -b)
const map = new Map()
for(let i = 0; i < removalArr.length; i++){
map.set(removalArr[i],i)
}
let res = 0
for(let i = 0; i < nums.length; i++){
res += map.get(nums[i])
}
return res
};