题意:
给你一个长度为n的数组,这个数组有(n-1)*n/2个子数组(1,2是1,2,-3的子数组,1,-3不是1,2,-3的子数组,不能中间缺一截)。问你这些子数组中,数组值的和为0的数组不是good数组,如果某个数组的某个子数组不是good数组,那么他也不是good数组。
(-1,2,-1,-3不是good数组,因为他的子数组-1,2,-1不是good数组)
问你所有的子数组里面,有多少个good数组(1≤n≤2×105,−109≤ai≤109)
输入:
3
1 2 -3
输出:
5
思路:要尝试(n-1)*n/2中情况,肯定是超时的,现在可以求以ai结尾的子数组有多少个是可以的。接下来求数组以ai结尾的的前缀和,假设前j个数的和跟前i个数的和一样(j<i),那么j到i这一段的值为0,由于只需要记录最接近的0,那么只需要记录最大的j,然后看代码你就懂了
#include <iostream>
#include <cstdio>
#include <iomanip>
#include <string>
#include <cstdlib>
#include <cstring>
#include <queue>
#include <set>
#include <vector>
#include <map>
#include <algorithm>
#include <cmath>
#include <stack>
#define INF 0x3f3f3f3f
#define LINF 0x3f3f3f3f3f3f3f3f
#define ll long long
#define ull unsigned long long
#define uint unsigned int
using namespace std;
ll k, s, q = -1;
int a, n;
int main() {
map<ll, ll> m;
scanf("%d", &n);
m[0] = 0;
for (int i = 1; i <= n; i++) {
scanf("%d", &a);
s += a;
if (m.count(s))
q = max(q, m[s]);
k += i - q - 1;
m[s] = i;
}
printf("%lld\n", k);
return 0;
}