Leetcode3139. 使数组中所有元素相等的最小开销

Every day a Leetcode

题目来源：3139. 使数组中所有元素相等的最小开销

解法1：分类讨论

情况1：如果 n≤2，那么只能用操作一，最小总开销为：(M−m)*c₁。

情况二：如果2*c₁<=c₂，用一次操作二不如用两次操作一，所以只需用操作一，把每个数都变成 M 的总开销是最小的。总共需要执行=(M−nums[0])+(M−nums[1])+⋯+(M−nums[n−1])=nM−(nums[0]+nums[1]+⋯+nums[n−1])次操作一。那么最小总开销为(nM−(nums[0]+nums[1]+⋯+nums[n−1]))*c₁。
​
情况三：如果2*c₁>c₂，我们尽量用第二种操作。

枚举目标数 g（由于每个数只能变大不能变小，所以 g≥l），那么第 i 个数需要增加 (g−nums[i])次。问题变为：

设所有桶加起来一共有 s 个石头，最多的桶有 m 个石头，结论为：

• 如果 m>s−m，则最多可以执行 (s−m) 次第二种操作，之后所有石头都只在最多的那个桶里，必须执行 (m−(s−m)) 次第一种操作。
• 如果 m≤s−m，则最多可以执行 ⌊s/2⌋ 次第二种操作，之后如果 s 是奇数那么还会剩下一个石头，执行一次第一种操作即可。

对于第二种情况，操作次数只和 s 有关，那么操作代价也只和 s 有关。所以如果 g=M 时已经是第二种情况了，那么继续增加 g 只会让 s 变得更大，没有意义。

剩下的问题就是，如果 g=M 时还是第一种情况，那么 g 增加到什么时候，一定会进入第二种情况呢？注意到 g 每次增加 1，s 会增加 n，而 m 只会增加 1，那么 m 和 (s−m) 之间的差距会缩小 (n−2)。因为现在讨论的情况中，n 最小是 3，那么 g 每增加 1，两者之间的差距就会缩小 1。由于 m≤s，那么 g 最多增加 m 就一定会进入第二种情况。由于 m≤M，我们只要枚举 g∈[M,2M] 即可。

代码：

class Solution
{
private:
	const int MOD = 1e9 + 7;

public:
	int minCostToEqualizeArray(vector<int> &nums, int cost1, int cost2)
	{
		int n = nums.size();
		int maxNum = *max_element(nums.begin(), nums.end());

		// 计算一开始每个桶里有几个石头
		long long diffSum = 0LL;
		int maxDiff = INT_MIN;
		for (int &x : nums)
		{
			int diff = maxNum - x;
			diffSum += diff;
			maxDiff = max(maxDiff, diff);
		}

		// 情况一
		if (n <= 2)
			return (diffSum * cost1) % MOD;
		// 情况二
		if (cost1 * 2 <= cost2)
			return (diffSum * cost1) % MOD;
        // 情况三
		long long ans = LLONG_MAX;
		// 从 maxNum 到 2 * maxNum 枚举目标数
		for (int i = maxNum; i <= 2 * maxNum; i++)
		{
			long long cost = 0LL;
			if (maxDiff > diffSum - maxDiff)
			{
				// 结论一
				cost += (diffSum - maxDiff) * cost2;
				cost += (maxDiff - (diffSum - maxDiff)) * cost1;
			}
			else
			{
				// 结论二
				cost = (diffSum / 2) * cost2;
				if (diffSum % 2 == 1)
					cost += cost1;
			}
			ans = min(ans, cost);
			// 目标数每增加 1，s 增加 n，m 增加 1
			diffSum += n;
			maxDiff++;
		}
		return ans % MOD;
	}
};

结果：

复杂度分析：

时间复杂度：O(n+M)，其中 n 是数组 nums 的元素个数，M 是数组 nums 的最大值。

空间复杂度：O(1)。