一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第28天,点击查看活动详情。
题目描述
这是代码源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;
}