【排序贪心】3107. 使数组中位数等于 K 的最少操作数|1604

原创已于 2024-10-13 18:39:09 修改 · 1k 阅读

16 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

困难算法题

于 2024-04-14 07:00:00 首次发布

文章介绍了如何使用贪心算法解决LeetCode3107问题，即在给定整数数组中调整元素使其中位数等于目标值K所需的最少操作次数。涉及到的知识点包括排序、二分图中的最大匹配、最小点覆盖和最大独立集的概念。

本文涉及知识点

C++贪心
排序

LeetCode 3107. 使数组中位数等于 K 的最少操作数

给你一个整数数组 nums 和一个非负整数 k 。一次操作中，你可以选择任一元素加 1 或者减 1 。
请你返回将 nums 中位数变为 k 所需要的最少操作次数。
一个数组的中位数指的是数组按非递减顺序排序后最中间的元素。如果数组长度为偶数，我们选择中间两个数的较大值为中位数。
示例 1：
输入：nums = [2,5,6,8,5], k = 4
输出：2
解释：我们将 nums[1] 和 nums[4] 减 1 得到 [2, 4, 6, 8, 4] 。现在数组的中位数等于 k 。
示例 2：
输入：nums = [2,5,6,8,5], k = 7
输出：3
解释：我们将 nums[1] 增加 1 两次，并且将 nums[2] 增加 1 一次，得到 [2, 7, 7, 8, 5] 。
示例 3：
输入：nums = [1,2,3,4,5,6], k = 4
输出：0
解释：数组中位数已经等于 k 了。

提示：
1 <= nums.length <= 2 * 10⁵
1 <= nums[i] <= 10⁹
1 <= k <= 10⁹

贪心

n = nums.length ，无轮n 是奇数，还是偶数。排序后，中为数的下标都是n/2。
$操作次数：\begin{cases} max(0,nums[i]-k) && i < n/2 \\ abs(nums[i]-k) && i == n/2 \\ max(0,k - nums[i]) && i > n/2 \\ \end{cases}$
除一个数为K外，n/2个数小于等于k，显然选择最小的n/2个数来操作成本最低。
n - n/2 -1 个数大于等于k，显然选择最大(n - n/2-1)来操作。

代码

class Solution {
public:
	long long minOperationsToMakeMedianK(vector<int>& nums, int k) {
		sort(nums.begin(), nums.end());
		int index = nums.size() / 2;
		int i = 0;
		long long ret = 0;
		for (; i < index; i++) {
			ret += max(0, nums[i] - k);
		}
		ret += abs(nums[i] - k);
		for (i++; i < nums.size(); i++) {
			ret += max(0, k - nums[i]);
		}
		return ret;
	}
};

扩展阅读

视频课程

有效学习：明确的目标及时的反馈拉伸区（难度合适），可以先学简单的课程，请移步CSDN学院，听白银讲师（也就是鄙人）的讲解。
https://edu.csdn.net/course/detail/38771

如何你想快速形成战斗了，为老板分忧，请学习C#入职培训、C++入职培训等课程
https://edu.csdn.net/lecturer/6176

我想对大家说的话
闻缺陷则喜是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛