【C++前缀和位运算贪心】2680. 最大或值|1912

闻缺陷则喜何志丹

于 2024-09-27 07:00:00 发布

阅读量251

点赞数 9

文章标签： c++ 算法力扣前缀和位运算最大或值

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

版权

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

96 篇文章 0 订阅

订阅专栏

本文涉及的基础知识点

C++算法：前缀和、前缀乘积、前缀异或的原理、源码及测试用例包括课程视频
 位运算、状态压缩、枚举子集汇总
贪心

LeetCode2680. 最大或值

给你一个下标从 0 开始长度为 n 的整数数组 nums 和一个整数 k 。每一次操作中，你可以选择一个数并将它乘 2 。
你最多可以进行 k 次操作，请你返回 nums[0] | nums[1] | … | nums[n - 1] 的最大值。
a | b 表示两个整数 a 和 b 的按位或运算。
示例 1：
输入：nums = [12,9], k = 1
输出：30
解释：如果我们对下标为 1 的元素进行操作，新的数组为 [12,18] 。此时得到最优答案为 12 和 18 的按位或运算的结果，也就是 30 。
示例 2：
输入：nums = [8,1,2], k = 2
输出：35
解释：如果我们对下标 0 处的元素进行操作，得到新数组 [32,1,2] 。此时得到最优答案为 32|1|2 = 35 。
提示：
1 <= nums.length <= 10⁵
1 <= nums[i] <= 10⁹
1 <= k <= 15

贪心+前缀和

贪心（决策包容性）：一定是将一个数操作k次。令nums 中为1的最高位是i1，则这样操作可以保证结果i1+k位为1。操作了两个数或更多的数，无法让第i+k位为1。
枚举操作了nums[i]，针对nums[i]，令x = 除nums[i]外,nums的异或和。则操作nums[i]的最大值cur = (nums[i] << k )|x 。
令cnt[i]记录nums[i]中1的数量。cnt1[i] 记录nums[i]第i为1的数量。
cnt[i]-cnt1[i] > 0 $\iff$ x的第i位是否是1。

代码

核心代码

	class Solution {
		public:
			long long maximumOr(vector<int>& nums, int k) {
				int cnt[30] = { 0 };
				for (const auto& n : nums) {
					for (int i = 0; i < 30; i++) {
						cnt[i] += bool((1 << i) & n);
					}
				}
				long long res = 0;
				for (const auto& n : nums) {
					int cnt1[30] = { 0 };
					memcpy(cnt1, cnt, sizeof(cnt));
					int x = 0;
					for (int i = 0; i < 30; i++) {
						cnt1[i] -=bool( (1 << i) & n);
						if (cnt1[i] > 0) {
							x |= (1 << i);
						}
					}
					long long cur = x | (((long long)n) << k );
					res = max(res, cur);
				}
				return res;
			}
		};;

核心代码

	vector<int> nums;
		int k;
		TEST_METHOD(TestMethod11)
		{
			nums = { 12, 9 }, k = 1;
			auto res = Solution().maximumOr(nums, k);
			AssertEx(30LL, res);
		}
		TEST_METHOD(TestMethod12)
		{
			nums = { 8,1,2 }, k = 2;
			auto res = Solution().maximumOr(nums, k);
			AssertEx(35LL, res);
		}

扩展阅读

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