#include <iostream> #include <vector> #include <map> #include <algorithm> using namespace std; void printv(const vector<int> &wyn) { for (auto i: wyn) cout << i << " "; cout << endl; } void mk_buc(const vector<int> &c,vector<vector<int>> &buc) { for (int i = 0; i < c.size(); i++) { buc.push_back({}); buc[i].push_back(c[i]); for (int j = 1; j < c.size() - i; j++) { buc[i].push_back(buc[i][j-1]+c[j+i]); } } } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, licz; vector<int> uc; vector<vector<int>> buc2; vector<int> buc; map<int,vector<int>> liczby; cin >> n; for (int i=0; i<n; i++) { cin >> licz; uc.push_back(licz); } mk_buc(uc, buc2); for (auto a: buc2) for(auto i: a) buc.push_back(i); for (int i = 0; i<buc.size(); i++) { liczby[buc[i]].push_back(i); } for (int i = 0; i<buc.size(); i++) { sort(liczby[buc[i]].begin(), liczby[buc[i]].end()); } int wynik = 0; int sum2 =0; auto ff = liczby[buc[0]].end(); for (int i = 0; i<buc.size(); i++) { for (int j = i+1; j<buc.size(); j++) { sum2 = -(buc[i]+buc[j]); ff = lower_bound(liczby[sum2].begin(), liczby[sum2].end(), j+1); if (ff != liczby[sum2].end()) { wynik+=(liczby[sum2].end()-ff); } } } cout << wynik << "\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 52 53 54 55 56 57 58 59 60 61 62 63 64 | #include <iostream> #include <vector> #include <map> #include <algorithm> using namespace std; void printv(const vector<int> &wyn) { for (auto i: wyn) cout << i << " "; cout << endl; } void mk_buc(const vector<int> &c,vector<vector<int>> &buc) { for (int i = 0; i < c.size(); i++) { buc.push_back({}); buc[i].push_back(c[i]); for (int j = 1; j < c.size() - i; j++) { buc[i].push_back(buc[i][j-1]+c[j+i]); } } } int main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int n, licz; vector<int> uc; vector<vector<int>> buc2; vector<int> buc; map<int,vector<int>> liczby; cin >> n; for (int i=0; i<n; i++) { cin >> licz; uc.push_back(licz); } mk_buc(uc, buc2); for (auto a: buc2) for(auto i: a) buc.push_back(i); for (int i = 0; i<buc.size(); i++) { liczby[buc[i]].push_back(i); } for (int i = 0; i<buc.size(); i++) { sort(liczby[buc[i]].begin(), liczby[buc[i]].end()); } int wynik = 0; int sum2 =0; auto ff = liczby[buc[0]].end(); for (int i = 0; i<buc.size(); i++) { for (int j = i+1; j<buc.size(); j++) { sum2 = -(buc[i]+buc[j]); ff = lower_bound(liczby[sum2].begin(), liczby[sum2].end(), j+1); if (ff != liczby[sum2].end()) { wynik+=(liczby[sum2].end()-ff); } } } cout << wynik << "\n"; } |