#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define st first
#define nd second
int main()
{
ios_base::sync_with_stdio(false);cin.tie(0);
ll n;
cin>>n;
vector<int>a(n);
for(int i=0;i<n;i++)
{
cin>>a[i];
if(i>0)
{
a[i]+=a[i-1];
}
}
vector<int>b(n*(n+1)/2);
for(int j=0;j<n;j++)
{
b[j]=a[j];
}
for(int i=1;i<n;i++)
{
for(int j=i;j<n;j++)
{
b[i*n-(i*(i-1)/2)+j-i]=a[j]-a[i-1];
}
}
n=b.size();
map<int,int>m;
ll wyn=0;
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
wyn+=m[(-1)*(b[i]+b[j])];
}
m[b[i]]++;
}
cout<<wyn<<endl;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | #include<bits/stdc++.h> using namespace std; #define ll long long #define st first #define nd second int main() { ios_base::sync_with_stdio(false);cin.tie(0); ll n; cin>>n; vector<int>a(n); for(int i=0;i<n;i++) { cin>>a[i]; if(i>0) { a[i]+=a[i-1]; } } vector<int>b(n*(n+1)/2); for(int j=0;j<n;j++) { b[j]=a[j]; } for(int i=1;i<n;i++) { for(int j=i;j<n;j++) { b[i*n-(i*(i-1)/2)+j-i]=a[j]-a[i-1]; } } n=b.size(); map<int,int>m; ll wyn=0; for(int i=0;i<n;i++) { for(int j=i+1;j<n;j++) { wyn+=m[(-1)*(b[i]+b[j])]; } m[b[i]]++; } cout<<wyn<<endl; } |
English