F. Eating Candies

双指针求最多相同糖果数量

最新推荐文章于 2025-12-01 22:20:41 发布

原创最新推荐文章于 2025-12-01 22:20:41 发布 · 143 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

算法错题本同时被 2 个专栏收录

71 篇文章

订阅专栏

玩转上号CF“游戏”

38 篇文章

订阅专栏

文章描述了一个使用双指针算法解决的编程问题，给定一个糖果数组，找到两个指针同时指向的糖果数量之和相同的最大数量。通过模拟操作，找到满足条件的指针位置，计算出吃掉相同数量糖果的最大次数。

题目：

样例：

输入
`4 3 10 20 10 6 2 1 4 2 4 1 5 1 2 4 8 16 9 7 3 20 5 15 1 11 8 10`

输出
2 6 0 7

思路：

思维题，模拟操作，在这里我们只要找到符合前面某个下标之前的总和等于后面开始往前遍历的总和，即可得出答案，有点像双指针算法，这里我们利用相减的特性，当它们吃糖果，吃到相等总和的时候，相差是 0 ，所以我们记录好相差是 0 的两个下标，然后取个答案 ans max 就可以得到吃的最多糖果总数量

代码详解如下：

#include <iostream>
#include <unordered_map>
#define endl '\n'
#define umap unordered_map
#pragma GCC optimize(3,"Ofast","inline")
#define ___G std::ios::sync_with_stdio(false),cin.tie(0), cout.tie(0)
using namespace std;
const int N = 2e6 + 10;
int n;
umap<int, int>a;	// 糖果数组

inline void solve()
{
	int ans = 0;
	int sum = 0;	// 记录吃糖果过程
	cin >> n;
	for (int i = 0; i < n; ++i)
		cin >> a[i];

	// 开始操作模拟
	// 这里用 i <= j 是因为 当我们数量为 2 的时候 比如  3  3  也是总共吃的数量相同
	for (int i = 0, j = n - 1; i <= j;)
	{
		// 左边先吃一颗，所以 >= 0    sum -=
		if (sum >= 0)
			sum -= a[i++];
		else
			sum += a[j--];

		// 当我们sum == 0 说明有可以吃的糖果相同数字量
		// 记录我们吃的总数量  n - j - 1  是我们倒数第几个   i 是我们当前左边吃的多少个
		if (!sum)
			ans = max(ans, i + n - j - 1);
	}
	cout << ans << endl;
}


int main()
{
//	freopen("a.txt", "r", stdin);
	___G;
	int _t = 1;
	cin >> _t;
	while (_t--)
	{
		solve();
	}

	return 0;
}

最后提交：