#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; } |