Eugene likes working with arrays. And today he needs your help in solving one challenging task.
An array c is a subarray of an array b if c can be obtained from b by deletion of several (possibly, zero or all) elements from the beginning and several (possibly, zero or all) elements from the end.
Let’s call a nonempty array good if for every nonempty subarray of this array, sum of the elements of this subarray is nonzero. For example, array [−1,2,−3] is good, as all arrays [−1], [−1,2], [−1,2,−3], [2], [2,−3], [−3] have nonzero sums of elements. However, array [−1,2,−1,−3] isn’t good, as his subarray [−1,2,−1] has sum of elements equal to 0.
Help Eugene to calculate the number of nonempty good subarrays of a given array a.
Input
The first line of the input contains a single integer n (1≤n≤2×105) — the length of array a.
The second line of the input contains n integers a1,a2,…,an (−109≤ai≤109) — the elements of a.
Output
Output a single integer — the number of good subarrays of a.
Examples
Input
3
1 2 -3
Output
5
Input
3
41 -41 41
Output
3
Note
In the first sample, the following subarrays are good: [1], [1,2], [2], [2,−3], [−3]. However, the subarray [1,2,−3] isn’t good, as its subarray [1,2,−3] has sum of elements equal to 0.
In the second sample, three subarrays of size 1 are the only good subarrays. At the same time, the subarray [41,−41,41] isn’t good, as its subarray [41,−41] has sum of elements equal to 0.
题意:给你n 个数的数组,让你能否从他的子数组中得到一个good数组,good数组是这样定义的:数组中所有元素的和不为0.让你求good数组的数量。
思路:前缀和+map,假设我们在a[i]处,以a[i]为结尾的子串就有i个,假设a[j]~a[k]的和为0,且j<k<=i,那么以a[i]为结尾的合法子串只有i-j个。我们处理一下前缀和,用map记录前缀和出现的位置,我们只要处理离i最近的那一个为0子串。
#include"cstdio"
#include"iostream"
#include"cstring"
#include"cmath"
#include"map"
#include"algorithm"
using namespace std;
#define maxn 200005
int a[maxn],b[maxn];
map<long long,int>mp;
int main()
{
int n;
cin>>n;
long long sum=0;
mp[0]=0;
int num=-1;
long long aum=0;
for(int i=1;i<=n;i++)
{
cin>>a[i];
sum+=a[i];
if(mp[sum]||sum==0)
{
num=max(num,mp[sum]);
}
//cout<<num;
aum=aum+i-num-1;
mp[sum]=i;
//cout<<mp[sum]<<endl;
}
cout<<aum<<endl;
return 0;
}