#include <bits/stdc++.h>
using namespace std;
vector<int> v;
map<int,int> ile;
void genBUC(int arr[],int n){
for(int i=1;i<n;i++){
arr[i]+=arr[i-1];
}
for (int i = 0; i < n; i++) {
if (ile[arr[i]] == 0) {
v.push_back(arr[i]);
}
ile[arr[i]]++;
}
int x;
for(int j=0;j<n-1;j++) {
for (int i = j+1; i < n; i++) {
x=arr[i]-arr[j];
if (ile[x] == 0) {
v.push_back(x);
}
ile[x]++;
}
}
}
int znajdz(){
int w=0;
int n=v.size();
if(n==1){
}
for(int i=0;i<n-1;i++){
int l=i+1;
int r=n-1;
while(l<r){
int s=v[i]+v[l]+v[r];
if(s==0){
w+=ile[v[i]]*ile[v[l]]*ile[v[r]];
l++;
}else if(s<0){
l++;
}else{
r--;
}
}
}
return w;
}
int main() {
cin.tie(0);
cout.tie(0);
ios_base::sync_with_stdio(0);
int n;
cin >> n;
int tab[n];
for(int i=0;i<n;i++){
cin >> tab[i];
}
genBUC(tab,n);
sort(v.begin(),v.end());
cout << znajdz();
}
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 52 53 54 55 56 57 58 59 60 61 62 | #include <bits/stdc++.h> using namespace std; vector<int> v; map<int,int> ile; void genBUC(int arr[],int n){ for(int i=1;i<n;i++){ arr[i]+=arr[i-1]; } for (int i = 0; i < n; i++) { if (ile[arr[i]] == 0) { v.push_back(arr[i]); } ile[arr[i]]++; } int x; for(int j=0;j<n-1;j++) { for (int i = j+1; i < n; i++) { x=arr[i]-arr[j]; if (ile[x] == 0) { v.push_back(x); } ile[x]++; } } } int znajdz(){ int w=0; int n=v.size(); if(n==1){ } for(int i=0;i<n-1;i++){ int l=i+1; int r=n-1; while(l<r){ int s=v[i]+v[l]+v[r]; if(s==0){ w+=ile[v[i]]*ile[v[l]]*ile[v[r]]; l++; }else if(s<0){ l++; }else{ r--; } } } return w; } int main() { cin.tie(0); cout.tie(0); ios_base::sync_with_stdio(0); int n; cin >> n; int tab[n]; for(int i=0;i<n;i++){ cin >> tab[i]; } genBUC(tab,n); sort(v.begin(),v.end()); cout << znajdz(); } |
English