当前位置：首页 > article >正文

【C++贪心】1262. 可被三整除的最大和|1762

article 2025/1/20 18:34:17

本文涉及知识点

C++贪心

LeetCode1262. 可被三整除的最大和

给你一个整数数组 nums，请你找出并返回能被三整除的元素最大和。
示例 1：
输入：nums = [3,6,5,1,8]
输出：18
解释：选出数字 3, 6, 1 和 8，它们的和是 18（可被 3 整除的最大和）。
示例 2：
输入：nums = [4]
输出：0
解释：4 不能被 3 整除，所以无法选出数字，返回 0。
示例 3：
输入：nums = [1,2,3,4,4]
输出：12
解释：选出数字 1, 3, 4 以及 4，它们的和是 12（可被 3 整除的最大和）。
提示：
1 <= nums.length <= 4 * 10⁴
1 <= nums[i] <= 10⁴

贪心

如果sum=nums之和是3的倍数，则返回之和。
如果sum%32 ，则删除最小的nums[i]%32 ，或删除最小或次小nums[i]%31。
如果sum%31，则删除最小的nums[i]%31 或删除最小或次小nums[i]%32。
如果排序，时间复杂度：O(nlogn)。
如果直接用nth或2轮选择排序，时间复杂度：O(n)

代码

核心代码

class Solution {
		public:
			int maxSumDivThree(vector<int>& nums) {				
				int sum = accumulate(nums.begin(), nums.end(), 0);
				if (0 == sum % 3) { return sum; }
				priority_queue<int> heap[3];
				for (const auto& n : nums) {
					heap[n % 3].emplace(n);
					if (heap[n % 3].size() > 2) { heap[n % 3].pop(); }
				}
				int del1 = 1'000'000, del2 = 1'000'000;
				const int inx = sum % 3;
				const int inx2 = 3 - inx;
				if (heap[inx].size()) { if (heap[inx].size() > 1) heap[inx].pop(); del1 = heap[inx].top(); }
				if (heap[inx2].size() >= 2) { del2 = heap[inx2].top(); heap[inx2].pop(); del2 += heap[inx2].top(); }
				return max(sum - del1, sum - del2);
			}
		};

单元测试

vector<int> nums;
		TEST_METHOD(TestMethod11)
		{
			nums = { 3,6,5,1,8 };
			auto res = Solution().maxSumDivThree(nums);
			AssertEx(18, res);
		}
		TEST_METHOD(TestMethod12)
		{
			nums = { 4 };
			auto res = Solution().maxSumDivThree(nums);
			AssertEx(0, res);
		}
		TEST_METHOD(TestMethod13)
		{
			nums = { 1,2,3,4,4 };
			auto res = Solution().maxSumDivThree(nums);
			AssertEx(12, res);
		}

扩展阅读

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