题目
#### [5777. 使数组元素相等的减少操作次数
https://leetcode-cn.com/problems/reduction-operations-to-make-the-array-elements-equal/
给你一个整数数组 nums
,你的目标是令 nums
中的所有元素相等。完成一次减少操作需要遵照下面的几个步骤:
- 找出
nums
中的 最大 值。记这个值为largest
并取其下标i
(下标从 0 开始计数)。如果有多个元素都是最大值,则取最小的i
。 - 找出
nums
中的 下一个最大 值,这个值 严格小于largest
,记为nextLargest
。 - 将
nums[i]
减少到nextLargest
。
返回使 nums
中的所有元素相等的操作次数。
示例 1:
输入:nums = [5,1,3]
输出:3
解释:需要 3 次操作使 nums 中的所有元素相等:
1. largest = 5 下标为 0 。nextLargest = 3 。将 nums[0] 减少到 3 。nums = [3,1,3] 。
2. largest = 3 下标为 0 。nextLargest = 1 。将 nums[0] 减少到 1 。nums = [1,1,3] 。
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 中的所有元素相等:
1. largest = 3 下标为 4 。nextLargest = 2 。将 nums[4] 减少到 2 。nums = [1,1,2,2,2] 。
2. largest = 2 下标为 2 。nextLargest = 1 。将 nums[2] 减少到 1 。nums = [1,1,1,2,2] 。
3. largest = 2 下标为 3 。nextLargest = 1 。将 nums[3] 减少到 1 。nums = [1,1,1,1,2] 。
4. largest = 2 下标为 4 。nextLargest = 1 。将 nums[4] 减少到 1 。nums = [1,1,1,1,1] 。
提示:
1 <= nums.length <= 5 * 104
1 <= nums[i] <= 5 * 104
题解
思路
先对数据进行排序,再根据元素的操作次数累加
代码
/**
* @param {number[]} nums
* @return {number}
*/
var reductionOperations = function(nums) {
const length = nums.length;
let ans = 0;
// 操作次数
let cnt = 0;
nums.sort((a, b) => a - b);
// 统计每个元素需要操作的次数
for(let i = 1; i < length; i++) {
// 如果和上一个值不相等,操作次数加1
if(nums[i] !== nums[i - 1]){
cnt++;
}
ans += cnt;
}
return ans;
};