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