【C++二分查找】2226. 每个小孩最多能分到多少糖果

最新推荐文章于 2024-09-15 11:40:06 发布

闻缺陷则喜何志丹

最新推荐文章于 2024-09-15 11:40:06 发布

阅读量656

点赞数 12

文章标签： c++ 算法力扣二分查找最多糖果相同

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

版权

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

77 篇文章 0 订阅

订阅专栏

本文涉及的基础知识点

C++二分查找

LeetCode2226. 每个小孩最多能分到多少糖果

给你一个下标从 0 开始的整数数组 candies 。数组中的每个元素表示大小为 candies[i] 的一堆糖果。你可以将每堆糖果分成任意数量的子堆，但无法再将两堆合并到一起。
另给你一个整数 k 。你需要将这些糖果分配给 k 个小孩，使每个小孩分到相同数量的糖果。每个小孩可以拿走至多一堆糖果，有些糖果可能会不被分配。
返回每个小孩可以拿走的最大糖果数目。
示例 1：
输入：candies = [5,8,6], k = 3
输出：5
解释：可以将 candies[1] 分成大小分别为 5 和 3 的两堆，然后把 candies[2] 分成大小分别为 5 和 1 的两堆。现在就有五堆大小分别为 5、5、3、5 和 1 的糖果。可以把 3 堆大小为 5 的糖果分给 3 个小孩。可以证明无法让每个小孩得到超过 5 颗糖果。
示例 2：
输入：candies = [2,5], k = 11
输出：0
解释：总共有 11 个小孩，但只有 7 颗糖果，但如果要分配糖果的话，必须保证每个小孩至少能得到 1 颗糖果。因此，最后每个小孩都没有得到糖果，答案是 0 。
提示：
1 <= candies.length <= 10⁵
1 <= candies[i] <= 10⁷
1 <= k <= 10¹²

二分查找

Cnt(mid)函数：拆分后，最多有多少堆数量大于等于mid。
二分查找类型：寻找尾端。
Check函数的参数范围：[0,10⁷]，一定有解。
Check函数：Cnt(mid) >= k
时间复杂度：O(mlogn)，n = candies.length,m=10⁷。

代码

核心代码

template<class INDEX_TYPE>
class CBinarySearch
{
public:
	CBinarySearch(INDEX_TYPE iMinIndex, INDEX_TYPE iMaxIndex):m_iMin(iMinIndex),m_iMax(iMaxIndex) {}
	template<class _Pr>
	INDEX_TYPE FindFrist( _Pr pr)
	{
		auto left = m_iMin - 1;
		auto rightInclue = m_iMax;
		while (rightInclue - left > 1)
		{
			const auto mid = left + (rightInclue - left) / 2;
			if (pr(mid))
			{
				rightInclue = mid;
			}
			else
			{
				left = mid;
			}
		}
		return rightInclue;
	}
	template<class _Pr>
	INDEX_TYPE FindEnd( _Pr pr)
	{
		int leftInclude = m_iMin;
		int right = m_iMax + 1;
		while (right - leftInclude > 1)
		{
			const auto mid = leftInclude + (right - leftInclude) / 2;
			if (pr(mid))
			{
				leftInclude = mid;
			}
			else
			{
				right = mid;
			}
		}
		return leftInclude;
	}
protected:
	const INDEX_TYPE m_iMin, m_iMax;
};

class Solution {
		public:
			int maximumCandies(vector<int>& candies, long long k) {
				auto Check = [&](int mid) {
					long long cnt = 0;
					for (const auto& n : candies) {
						cnt += n / mid;
					}
					return cnt >= k;
				};
				return CBinarySearch(0, 10'000'000).FindEnd(Check);
			}
		};

单元测试

	vector<int>candies; long long k;
		TEST_METHOD(TestMethod11)
		{
			candies = { 5,8,6 }, k = 3;
			auto res = Solution().maximumCandies(candies, k);
			AssertEx(5, res);
		}
		TEST_METHOD(TestMethod12)
		{
			candies = { 2,5 }, k = 11;
			auto res = Solution().maximumCandies(candies, k);
			AssertEx(0, res);
		}

扩展阅读

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