代码源:863、吃糖果

222 阅读2分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第28天,点击查看活动详情logo.png

题目描述

这是代码源4月29日的div2每日一题

吃糖果 - 题目 - Daimayuan Online Judge

桌子上从左到右放着 n 个糖果。糖果从左到右编号,第 i 块糖果的重量为 wi。小明和小红要吃糖果。

小明从左边开始吃任意数量的糖果。(连续吃,不能跳过糖果)

小红从右边开始吃任意数量的糖果。(连续吃,不能跳过糖果)

当然,如果小明吃了某个糖果,小红就不能吃它(反之亦然)。

他们的目标是吃同样重量的糖果,请问此时他们总共最多能吃多少个糖果?

输入格式

第一行包含一个整数 n,表示桌上糖果的数量。

第二行包含 n 个整数 w1,w2,…,wn,表示糖果从左到右的重量。

输出格式

一个整数,表示小明和小红在满足条件的情况下总共可以吃的糖果的最大数量。

数据范围

1≤n≤2∗10^5,1≤wi≤10^4

输入样例

9
7 3 20 5 15 1 11 8 10

输出样例

7

问题解析

这题翻译成人话就是,两个人一个从左往右吃糖果,一个从右往左吃糖果,他们俩人吃的糖果重量要一样,问他们两人最多可以吃多少糖果。其实就是说,从左往右的连续和从右往左的连续和相等的情况下,这两个连续和一共用了多少个数(数量小于等于n)。

那我们可以从左往右先算一遍前缀和,用哈希表记录得到这个前缀和用了多少数,然后再从右往左算一遍后缀和,每次看一下计算的后缀和是否有在哈希表中出现过,如果出现过,就更新一下个数的最大值。注意,当两个序列的个数和大于n时就要结束计算了。

#include<iostream>
using namespace std;
#include<vector>
#include<algorithm>
#include<math.h>
#include<set>
#include<numeric>
#include<string>
#include<string.h>
#include<iterator>
#include<map>
#include<unordered_map>
#include<stack>
#include<list>
#include<queue>
#include<iomanip>

#define endl '\n';
typedef long long ll;
typedef pair<ll, ll>PII;

inline int read() {
    int x = 0; char ch = getchar();
    while (ch < '0' || ch > '9') ch = getchar();
    while (ch >= '0' && ch <= '9') x = (x << 3) + (x << 1) + ch - '0', ch = getchar();
    return x;
}

inline void write(int x) {
    if (x > 9) write(x / 10);
    putchar(x % 10 | '0');
}

int main()
{
	int n;
	n=read();
	map<ll, int>alice, bob;
	ll sum = 0, len = 0;
	vector<ll>v(n);
	for (int i = 0; i < n; i++)
	{
		v[i]=read();
		sum += v[i];
		alice[sum] = i + 1;
	}
	sum = 0;
	for (int i = n - 1; i >= 0; i--)
	{
		sum += v[i];
		if (alice[sum] != 0 && alice[sum] + n - i <= n)
		{
			len = alice[sum] + n - i;
		}
	}
	write(len);
    return 0;
}