题意:
题目链接:http://hihocoder.com/problemset/problem/1534
一串长度为n的序列,分成三段,要求三段数字总和两两之间差的绝对值不超过1,问一共有多少种不同的分割方案。
思路:
只需要分成三段,那么枚举第一刀的位置,然后再查询之后有哪些位置分割之后,三段和可以满足条件即可,利用map维护前缀和的数目就行了。
代码:
#include <cstdio>
#include <iostream>
#include <cstring>
#include <vector>
#include <map>
#include <cmath>
using namespace std;
typedef long long LL;
const int MAXN = 1e5 + 10;
LL a[MAXN], sum[MAXN];
int main() {
//freopen("in.txt", "r", stdin);
int n;
scanf("%d", &n);
map <LL, LL> mp;
for (int i = 1; i <= n; i++) {
scanf("%lld", &a[i]);
sum[i] = sum[i - 1] + a[i];
mp[sum[i]]++;
}
LL ans = 0;
mp[sum[n]]--;
for (int i = 1; i < n - 1; i++) {
mp[sum[i]]--;
for (int j = -1; j <= 1; j++) {
LL x = sum[i], y = sum[i] + j, z = sum[n] - y - x;
if (abs(x - y) <= 1 && abs(x - z) <= 1 && abs(y - z) <= 1) {
ans += mp[sum[i] + y];
}
}
}
printf("%lld\n", ans);
return 0;
}