题目:![](https://img-blog.csdnimg.cn/a5aeb41d1402414a8e5e46d2c0a15128.png)
样例:
|
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;
}
最后提交: