#include <bits/stdc++.h> using namespace std; #define ll long long struct suma2 { int pierwszy; int drugi; int suma; }; bool mysort(suma2 a, suma2 b) { return a.suma < b.suma; } int findIndex(vector<suma2> vec, int sz) { int left = 0, right = vec.size() - 1, result = -1; while (left <= right) { int mid = left + (right - left) / 2; if (vec[mid].suma == sz) { result = mid; right = mid - 1; } else if (vec[mid].suma < sz) left = mid + 1; else right = mid - 1; } return result; } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n; cin >> n; int ucb[n]; for (int i=0; i<n; i++) { int a; cin >> a; ucb[i] = a; } vector<int> bucb; for (int i=0; i<n; i++) { int maxDl = n-i; for (int j=1; j<=maxDl; j++) { int sum = 0; for (int k=0; k<j; k++) { sum += ucb[i+k]; } bucb.push_back(sum); } } vector<suma2> suma2Vec; for (int i=0; i<bucb.size(); i++) { for (int j=i+1; j<bucb.size(); j++) { int sum = bucb[i] + bucb[j]; suma2 s; s.pierwszy = i; s.drugi = j; s.suma = sum; suma2Vec.push_back(s); } } sort(suma2Vec.begin(), suma2Vec.end(), mysort); ll cnt = 0; for (int i=0; i<bucb.size(); i++) { int a = bucb[i]; int szukanaSuma = -a; int idxBeg = findIndex(suma2Vec, szukanaSuma); if (idxBeg != -1) { while (suma2Vec[idxBeg].suma == szukanaSuma) { suma2 s = suma2Vec[idxBeg]; if (s.pierwszy < s.drugi && s.drugi < i) { cnt++; } idxBeg++; } } } cout << cnt; return 0; }
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 71 72 73 74 75 76 77 78 79 80 81 82 | #include <bits/stdc++.h> using namespace std; #define ll long long struct suma2 { int pierwszy; int drugi; int suma; }; bool mysort(suma2 a, suma2 b) { return a.suma < b.suma; } int findIndex(vector<suma2> vec, int sz) { int left = 0, right = vec.size() - 1, result = -1; while (left <= right) { int mid = left + (right - left) / 2; if (vec[mid].suma == sz) { result = mid; right = mid - 1; } else if (vec[mid].suma < sz) left = mid + 1; else right = mid - 1; } return result; } int main(){ ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n; cin >> n; int ucb[n]; for (int i=0; i<n; i++) { int a; cin >> a; ucb[i] = a; } vector<int> bucb; for (int i=0; i<n; i++) { int maxDl = n-i; for (int j=1; j<=maxDl; j++) { int sum = 0; for (int k=0; k<j; k++) { sum += ucb[i+k]; } bucb.push_back(sum); } } vector<suma2> suma2Vec; for (int i=0; i<bucb.size(); i++) { for (int j=i+1; j<bucb.size(); j++) { int sum = bucb[i] + bucb[j]; suma2 s; s.pierwszy = i; s.drugi = j; s.suma = sum; suma2Vec.push_back(s); } } sort(suma2Vec.begin(), suma2Vec.end(), mysort); ll cnt = 0; for (int i=0; i<bucb.size(); i++) { int a = bucb[i]; int szukanaSuma = -a; int idxBeg = findIndex(suma2Vec, szukanaSuma); if (idxBeg != -1) { while (suma2Vec[idxBeg].suma == szukanaSuma) { suma2 s = suma2Vec[idxBeg]; if (s.pierwszy < s.drugi && s.drugi < i) { cnt++; } idxBeg++; } } } cout << cnt; return 0; } |