#include<bits/stdc++.h> #include<ext/pb_ds/assoc_container.hpp> #include<ext/pb_ds/tree_policy.hpp> #define ST first #define ND second #define ll long long #define ld long double #define imie(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] " using namespace std; using namespace __gnu_pbds; // replace ll with pair if multiset needed typedef tree<ll, null_type, greater<ll>, rb_tree_tag, tree_order_statistics_node_update> ordered_set; ordered_set secik; const ll INF = 1e9 + 9; const ll MOD = 1e9 + 7; const long long LINF = (ll)1e18 + 3; // secik.insert({x, t++}); // secik.erase(secik.lower_bound({x,-1})); // *secik.find_by_order(x)).first << "\n"; // secik.order_of_key(x) //random_device device; //mt19937 gener(device()); //uniform_int_distribution<ll > gen(0,n-1); //gen(gener); // generate random number int n; ll getMedian(){ int x = secik.size(); if(secik.size() % 2 == 1){ return 2*(*secik.find_by_order(x/2)); } else { return (*secik.find_by_order(x/2) + *secik.find_by_order(x/2 - 1)); } } bool good(){ return (secik.size() + getMedian()) == 2*n+1; } int main(){ ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); cin >> n; vector<int> tab(n); ll ans = 0; for(int i = 0; i < n; i++) cin >> tab[i]; for(int len = 1; len <= n; len++){ secik.clear(); for(int i = 0; i < len; i++) secik.insert(tab[i]); //cout << getMedian() << " " << endl; if(good()) ans++; for(int i = len; i < n; i++){ secik.insert(tab[i]); secik.erase(tab[i-len]); if(good()) ans++; } } cout << 2*n+1 << " " << ans << "\n"; 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 | #include<bits/stdc++.h> #include<ext/pb_ds/assoc_container.hpp> #include<ext/pb_ds/tree_policy.hpp> #define ST first #define ND second #define ll long long #define ld long double #define imie(...) " [" << #__VA_ARGS__ ": " << (__VA_ARGS__) << "] " using namespace std; using namespace __gnu_pbds; // replace ll with pair if multiset needed typedef tree<ll, null_type, greater<ll>, rb_tree_tag, tree_order_statistics_node_update> ordered_set; ordered_set secik; const ll INF = 1e9 + 9; const ll MOD = 1e9 + 7; const long long LINF = (ll)1e18 + 3; // secik.insert({x, t++}); // secik.erase(secik.lower_bound({x,-1})); // *secik.find_by_order(x)).first << "\n"; // secik.order_of_key(x) //random_device device; //mt19937 gener(device()); //uniform_int_distribution<ll > gen(0,n-1); //gen(gener); // generate random number int n; ll getMedian(){ int x = secik.size(); if(secik.size() % 2 == 1){ return 2*(*secik.find_by_order(x/2)); } else { return (*secik.find_by_order(x/2) + *secik.find_by_order(x/2 - 1)); } } bool good(){ return (secik.size() + getMedian()) == 2*n+1; } int main(){ ios::sync_with_stdio(false); cin.tie(0); cout.tie(0); cin >> n; vector<int> tab(n); ll ans = 0; for(int i = 0; i < n; i++) cin >> tab[i]; for(int len = 1; len <= n; len++){ secik.clear(); for(int i = 0; i < len; i++) secik.insert(tab[i]); //cout << getMedian() << " " << endl; if(good()) ans++; for(int i = len; i < n; i++){ secik.insert(tab[i]); secik.erase(tab[i-len]); if(good()) ans++; } } cout << 2*n+1 << " " << ans << "\n"; return 0; } |