【C++二分查找前缀和】2602. 使数组元素全部相等的最少操作次数|1903

闻缺陷则喜何志丹

已于 2024-09-08 18:02:13 修改

阅读量675

点赞数 22

文章标签： c++ 力扣算法二分查找前缀和最少操作次数

于 2024-08-29 15:48:19 首次发布

本文链接：https://blog.csdn.net/he_zhidan/article/details/141376649

版权

工作级难度算法题解专栏收录该内容

110 篇文章 0 订阅

订阅专栏

本文涉及的基础知识点

C++二分查找
 C++算法：前缀和、前缀乘积、前缀异或的原理、源码及测试用例包括课程视频

LeetCode2602. 使数组元素全部相等的最少操作次数

同时给你一个长度为 m 的整数数组 queries 。第 i 个查询中，你需要将 nums 中所有元素变成 queries[i] 。你可以执行以下操作任意次：
将数组里一个元素增大或者减小 1 。
请你返回一个长度为 m 的数组 answer ，其中 answer[i]是将 nums 中所有元素变成 queries[i] 的最少操作次数。
注意，每次查询后，数组变回最开始的值。
示例 1：
输入：nums = [3,1,6,8], queries = [1,5]
输出：[14,10]
解释：第一个查询，我们可以执行以下操作：

将 nums[0] 减小 2 次，nums = [1,1,6,8] 。
将 nums[2] 减小 5 次，nums = [1,1,1,8] 。
将 nums[3] 减小 7 次，nums = [1,1,1,1] 。
第一个查询的总操作次数为 2 + 5 + 7 = 14 。
第二个查询，我们可以执行以下操作：
将 nums[0] 增大 2 次，nums = [5,1,6,8] 。
将 nums[1] 增大 4 次，nums = [5,5,6,8] 。
将 nums[2] 减小 1 次，nums = [5,5,5,8] 。
将 nums[3] 减小 3 次，nums = [5,5,5,5] 。
第二个查询的总操作次数为 2 + 4 + 1 + 3 = 10 。
示例 2：

输入：nums = [2,9,6,3], queries = [10]
输出：[20]
解释：我们可以将数组中所有元素都增大到 10 ，总操作次数为 8 + 1 + 4 + 7 = 20 。

提示：
n == nums.length
m == queries.length
1 <= n, m <= 10⁵
1 <= nums[i], queries[i] <= 10⁹

C++二分查找

n = nums.size()
对num按升序排序，令nums[0…j-1]都小于que[i]， nums[j…]都大于等于que[i]。
则结果是：que[i] $\times$ j - nums[0…j-1]之和 + nums[j…]之和- que[i] $\times$ (n-j)
求子数组之和用前缀和。

代码

核心代码

class Solution {
		public:
			vector<long long> minOperations(vector<int>& nums, vector<int>& queries) {
				sort(nums.begin(), nums.end());
				vector<long long> preSum(1);
				for (const auto& n : nums) {
					preSum.emplace_back(n + preSum.back());
				}
				vector<long long> ret;
				for (const long long que : queries) {
					int j = lower_bound(nums.begin(), nums.end(), que) - nums.begin();
					const long long cur =que * j - preSum[j] + (preSum.back() - preSum[j]) - que * (nums.size() - j);
					ret.emplace_back(cur);
				}
				return ret;
			}
		};

单元测试

vector<int> nums, queries;
		TEST_METHOD(TestMethod11)
		{
			nums = { 3, 1, 6, 8 }, queries = { 1, 5 };
			auto res = Solution().minOperations(nums, queries);
			AssertEx({ 14,10 }, res);
		}
		TEST_METHOD(TestMethod12)
		{
			nums = { 2,9,6,3 }, queries = {10 };
			auto res = Solution().minOperations(nums, queries);
			AssertEx({ 20 }, res);
		}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功