class Solution {
public:
int reductionOperations(vector<int>& nums) {
}
};
1887. 使数组元素相等的减少操作次数
给你一个整数数组 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 * 1041 <= nums[i] <= 5 * 104原站题解
golang 解法, 执行用时: 172 ms, 内存消耗: 8.2 MB, 提交时间: 2022-11-27 12:51:16
func reductionOperations(nums []int) int {
sort.Ints(nums)
n, ans, cnt := len(nums), 0, 0
for i := 1; i < n; i++ {
if nums[i] != nums[i-1] {
cnt++
}
ans += cnt
}
return ans
}
cpp 解法, 执行用时: 168 ms, 内存消耗: 81 MB, 提交时间: 2022-11-27 12:49:22
class Solution {
public:
int reductionOperations(vector<int>& nums) {
sort(nums.begin(),nums.end(),[&](int a,int b){return a>b;});
int ans = 0;
for(int i=0;i<nums.size()-1;i++)
{
if(nums[i]>nums[i+1]) ans+=i+1;
}
return ans;
}
};
java 解法, 执行用时: 8 ms, 内存消耗: 50.1 MB, 提交时间: 2022-11-27 12:49:03
class Solution {
public int reductionOperations(int[] nums) {
int[] hash = new int[50001];
int min = Integer.MAX_VALUE;
for (int i = 0; i < nums.length; i++) {
hash[nums[i]]++;
min = Math.min(min, nums[i]);
}
int ans = 0, sum = 0;
for (int i = hash.length - 1; i >= 0; i--) {
if (hash[i] != 0 && i != min) {
sum += hash[i];
ans += sum;
}
}
return ans;
}
}
python3 解法, 执行用时: 232 ms, 内存消耗: 21 MB, 提交时间: 2022-11-27 12:48:40
class Solution:
def reductionOperations(self, nums: List[int]) -> int:
nums.sort()
n = len(nums)
res = 0 # 总操作次数
cnt = 0 # 每个元素操作次数
for i in range(1, n):
if nums[i] != nums[i-1]:
cnt += 1
res += cnt
return res