#include<bits/stdc++.h> using namespace std; int main(){ int n; cin >> n; int liczba; int sumy[n]; int suma = 0; for(int i = 1; i <= n; i++){ cin >> liczba; suma += liczba; sumy[i] = suma; } sumy[0] = 0; vector <int> ciag; int izer = 0; for(int i = 0; i <= n; i++){ for(int i2 = i + 1; i2 <= n; i2++){ ciag.push_back(sumy[i2] - sumy[i]); if(sumy[i2] - sumy[i] == 0){ izer ++; } } } int zmiana = -1; sort(ciag.begin(), ciag.end()); for(int i = 0; i < ciag.size(); i++){ if(ciag[i] >= 0){ zmiana = i; break; } } if(zmiana == -1){ cout << 0; } else{ long long wynik = 0; for(int i = 0; i < zmiana; i++){ for(int i2 = i + 1; i2 < zmiana; i2++){ for(int x = zmiana; x < ciag.size(); x++){ if(ciag[i] + ciag[i2] + ciag[x] == 0){ wynik++; } else if(ciag[i] + ciag[i2] + ciag[x] > 0){ break; } } } } for(int i = zmiana; i < ciag.size(); i++){ for(int i2 = i + 1; i2 < ciag.size(); i2++){ for(int x = 0; x < zmiana; x++){ if(ciag[i] + ciag[i2] + ciag[x] == 0){ wynik++; } else if(ciag[i] + ciag[i2] + ciag[x] > 0){ break; } } } } if(izer > 2){ wynik += izer*(izer - 1)*(izer - 2) / 6; } cout << wynik; } }
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 63 64 65 66 67 68 69 70 | #include<bits/stdc++.h> using namespace std; int main(){ int n; cin >> n; int liczba; int sumy[n]; int suma = 0; for(int i = 1; i <= n; i++){ cin >> liczba; suma += liczba; sumy[i] = suma; } sumy[0] = 0; vector <int> ciag; int izer = 0; for(int i = 0; i <= n; i++){ for(int i2 = i + 1; i2 <= n; i2++){ ciag.push_back(sumy[i2] - sumy[i]); if(sumy[i2] - sumy[i] == 0){ izer ++; } } } int zmiana = -1; sort(ciag.begin(), ciag.end()); for(int i = 0; i < ciag.size(); i++){ if(ciag[i] >= 0){ zmiana = i; break; } } if(zmiana == -1){ cout << 0; } else{ long long wynik = 0; for(int i = 0; i < zmiana; i++){ for(int i2 = i + 1; i2 < zmiana; i2++){ for(int x = zmiana; x < ciag.size(); x++){ if(ciag[i] + ciag[i2] + ciag[x] == 0){ wynik++; } else if(ciag[i] + ciag[i2] + ciag[x] > 0){ break; } } } } for(int i = zmiana; i < ciag.size(); i++){ for(int i2 = i + 1; i2 < ciag.size(); i2++){ for(int x = 0; x < zmiana; x++){ if(ciag[i] + ciag[i2] + ciag[x] == 0){ wynik++; } else if(ciag[i] + ciag[i2] + ciag[x] > 0){ break; } } } } if(izer > 2){ wynik += izer*(izer - 1)*(izer - 2) / 6; } cout << wynik; } } |