#include <bits/stdc++.h>
using namespace std;
//#define int long long
#define ll long long
#define ld long double
#define pb push_back
#define nd second
#define st first
#define sz size
#define forr(i, n) for(int i=1;i<=n;i++)
const ll infl=1e18+90;
const int inf=1e9+90;
const int roz=509;
const int roz2=250000;
const int roz3=2e7;
int S[2*roz3+93];
int tab[roz], tab2[roz2], pary[roz2];
signed main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int n, wny=0;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>tab[i];
}
int licz=0;
for(int i=1;i<=n;i++)
{
int sum=0;
for(int j=i;j<=n;j++)
{
sum+=tab[j];
tab2[++licz]=sum;
}
}
n=licz;
licz=0;
int wyn=0;
for(int i=1;i<=n;i++)
{
wyn+=S[-tab2[i]+roz3];
for(int j=1;j<i;j++) S[tab2[i]+tab2[j]+roz3]++;
}
cout<<wyn<<"\n";
}
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 45 46 47 48 49 50 51 | #include <bits/stdc++.h> using namespace std; //#define int long long #define ll long long #define ld long double #define pb push_back #define nd second #define st first #define sz size #define forr(i, n) for(int i=1;i<=n;i++) const ll infl=1e18+90; const int inf=1e9+90; const int roz=509; const int roz2=250000; const int roz3=2e7; int S[2*roz3+93]; int tab[roz], tab2[roz2], pary[roz2]; signed main() { ios_base::sync_with_stdio(false); cin.tie(NULL); int n, wny=0; cin>>n; for(int i=1;i<=n;i++) { cin>>tab[i]; } int licz=0; for(int i=1;i<=n;i++) { int sum=0; for(int j=i;j<=n;j++) { sum+=tab[j]; tab2[++licz]=sum; } } n=licz; licz=0; int wyn=0; for(int i=1;i<=n;i++) { wyn+=S[-tab2[i]+roz3]; for(int j=1;j<i;j++) S[tab2[i]+tab2[j]+roz3]++; } cout<<wyn<<"\n"; } |
English