// #ifndef __SIZEOF_INT128__ #define __SIZEOF_INT128__ #endif #pragma GCC optimize("Ofast") #include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> using namespace std; using namespace chrono; using namespace __gnu_pbds; template <typename T> using oset = tree<T, null_type, less_equal<T>, rb_tree_tag, tree_order_statistics_node_update>; #define rep(i, p, k) for(int i(p); i < (k); ++i) #define per(i, p, k) for(int i(p); i > (k); --i) #define sz(x) (int)(x).size() #define sc static_cast typedef long long ll; typedef long double ld; typedef unsigned int uint; typedef unsigned long long ull; typedef __int128_t lll; //#define int ll template <typename T = int> using par = std::pair <T, T>; #define fi first #define se second #define test int _number_of_tests(in()); while(_number_of_tests--) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define pb emplace_back struct Timer { string name{""}; time_point<high_resolution_clock> end, start{high_resolution_clock::now()}; duration<float, std::milli> dur; Timer() = default; Timer(string nm): name(nm) {} ~Timer() { end = high_resolution_clock::now(); dur= end - start; cout << "@" << name << "> " << dur.count() << " ms" << '\n'; } }; template <typename T = int> inline T in() { static T x; std::cin >> x; return x; } std::string yn(bool b) { if(b) return "YES\n"; else return "NO\n"; } template <typename F, typename S> std::ostream& operator<<(std::ostream& out, const std::pair <F, S>& par); template <typename T> std::ostream& operator<< (std::ostream& out, const std::vector <T>& wek) { for(const auto& i : wek)out << i << ' '; return out; } template <typename F, typename S> std::ostream& operator<<(std::ostream& out, const std::pair <F, S>& par) { out << '{'<<par.first<<", "<<par.second<<"}"; return out; } #define show(x) cerr << #x << " = " << x << '\n'; constexpr int MAXNM = 1e6 + 3; ll o[MAXNM]; int f(vector <int> a, int m) //nlog { if(sz(a) == 1)return 2; int n(sz(a)); vector <int> v; bool bk(a[0] < a[1]); int c(2); rep(i, 2, n){ if(bk != (a[i-1] < a[i])){ v.pb(c); c = 1; } ++c; bk = a[i-1] < a[i]; } v.pb(c); int p(2), k((n+2)/2), s; while(k > p){ s = (p+k)/2; int dd(m); for(auto i: v){ int j(0); while((i+2*j)/(j+1) > s){ ++j; --dd; } } if(dd < 0)p = s+1; else k = s; } return p; } void add(vector <int> a, int l, int m) { if(sz(a) == 1){ if(l)o[2] += m; return; } int n(sz(a)); vector <int> v; bool bk(a[0] < a[1]); int c(2); rep(i, 2, n){ if(bk != (a[i-1] < a[i])){ v.pb(c); c = 1; } ++c; bk = a[i-1] < a[i]; } v.pb(c); // cerr << a << " -> " << v << ", " << l << '\n'; n = sz(v); vector <int> dz(n, 0); priority_queue <pair <int, int>> kul; rep(i, 0, n)kul.push(par<int>{v[i], i}); rep(i, 1, l+1){ auto [w, g] = kul.top(); kul.pop(); ++dz[g]; kul.push(par<int>{(v[g]+dz[g]*2)/(dz[g]+1), g}); // cerr << i << " => " << kul.top().first << '\n'; o[kul.top().first] += m-i+1; } } std::int32_t main() { std::cout.tie(nullptr); //for luck std::cin.tie(nullptr); std::ios_base::sync_with_stdio(0); int n(in()), m(in()); vector <int> a(n), b(m); rep(i, 0, n)cin >> a[i]; rep(i, 0, m)cin >> b[i]; rep(i, 0, n)rep(j, i, n){ vector <int> v; rep(k, i, j+1)v.pb(a[k]); add(v, min(m, sz(v)), m); } rep(i, 0, m)rep(j, i, m){ vector <int> v; rep(k, i, j+1)v.pb(b[k]); add(v, min(n, sz(v)-1), n); } rep(i, 1, n+m+1)cout << o[i] << ' '; cout << '\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 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 | // #ifndef __SIZEOF_INT128__ #define __SIZEOF_INT128__ #endif #pragma GCC optimize("Ofast") #include <bits/stdc++.h> #include <ext/pb_ds/assoc_container.hpp> #include <ext/pb_ds/tree_policy.hpp> using namespace std; using namespace chrono; using namespace __gnu_pbds; template <typename T> using oset = tree<T, null_type, less_equal<T>, rb_tree_tag, tree_order_statistics_node_update>; #define rep(i, p, k) for(int i(p); i < (k); ++i) #define per(i, p, k) for(int i(p); i > (k); --i) #define sz(x) (int)(x).size() #define sc static_cast typedef long long ll; typedef long double ld; typedef unsigned int uint; typedef unsigned long long ull; typedef __int128_t lll; //#define int ll template <typename T = int> using par = std::pair <T, T>; #define fi first #define se second #define test int _number_of_tests(in()); while(_number_of_tests--) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define pb emplace_back struct Timer { string name{""}; time_point<high_resolution_clock> end, start{high_resolution_clock::now()}; duration<float, std::milli> dur; Timer() = default; Timer(string nm): name(nm) {} ~Timer() { end = high_resolution_clock::now(); dur= end - start; cout << "@" << name << "> " << dur.count() << " ms" << '\n'; } }; template <typename T = int> inline T in() { static T x; std::cin >> x; return x; } std::string yn(bool b) { if(b) return "YES\n"; else return "NO\n"; } template <typename F, typename S> std::ostream& operator<<(std::ostream& out, const std::pair <F, S>& par); template <typename T> std::ostream& operator<< (std::ostream& out, const std::vector <T>& wek) { for(const auto& i : wek)out << i << ' '; return out; } template <typename F, typename S> std::ostream& operator<<(std::ostream& out, const std::pair <F, S>& par) { out << '{'<<par.first<<", "<<par.second<<"}"; return out; } #define show(x) cerr << #x << " = " << x << '\n'; constexpr int MAXNM = 1e6 + 3; ll o[MAXNM]; int f(vector <int> a, int m) //nlog { if(sz(a) == 1)return 2; int n(sz(a)); vector <int> v; bool bk(a[0] < a[1]); int c(2); rep(i, 2, n){ if(bk != (a[i-1] < a[i])){ v.pb(c); c = 1; } ++c; bk = a[i-1] < a[i]; } v.pb(c); int p(2), k((n+2)/2), s; while(k > p){ s = (p+k)/2; int dd(m); for(auto i: v){ int j(0); while((i+2*j)/(j+1) > s){ ++j; --dd; } } if(dd < 0)p = s+1; else k = s; } return p; } void add(vector <int> a, int l, int m) { if(sz(a) == 1){ if(l)o[2] += m; return; } int n(sz(a)); vector <int> v; bool bk(a[0] < a[1]); int c(2); rep(i, 2, n){ if(bk != (a[i-1] < a[i])){ v.pb(c); c = 1; } ++c; bk = a[i-1] < a[i]; } v.pb(c); // cerr << a << " -> " << v << ", " << l << '\n'; n = sz(v); vector <int> dz(n, 0); priority_queue <pair <int, int>> kul; rep(i, 0, n)kul.push(par<int>{v[i], i}); rep(i, 1, l+1){ auto [w, g] = kul.top(); kul.pop(); ++dz[g]; kul.push(par<int>{(v[g]+dz[g]*2)/(dz[g]+1), g}); // cerr << i << " => " << kul.top().first << '\n'; o[kul.top().first] += m-i+1; } } std::int32_t main() { std::cout.tie(nullptr); //for luck std::cin.tie(nullptr); std::ios_base::sync_with_stdio(0); int n(in()), m(in()); vector <int> a(n), b(m); rep(i, 0, n)cin >> a[i]; rep(i, 0, m)cin >> b[i]; rep(i, 0, n)rep(j, i, n){ vector <int> v; rep(k, i, j+1)v.pb(a[k]); add(v, min(m, sz(v)), m); } rep(i, 0, m)rep(j, i, m){ vector <int> v; rep(k, i, j+1)v.pb(b[k]); add(v, min(n, sz(v)-1), n); } rep(i, 1, n+m+1)cout << o[i] << ' '; cout << '\n'; return 0; } |