Untitled
unknown
plain_text
2 years ago
7.4 kB
5
Indexable
#include <bits/stdc++.h> #define all(x) x.begin(), x.end() #define r_all(x) x.rbegin(), x.rend() #define sz(x)(ll) x.size() #define g_max(x, y) x = max(x, y) #define g_min(x, y) x = min(x, y) #define rsz(a, n) a.resize(n) #define ass(a, n) a.assign(n, 0) #define YES() cout << "YES\n" #define Yes cout << "Yes\n" #define NO() cout << "NO\n" #define No() cout << "No\n" #define endl "\n" #define print(a) for (auto &x: a) cout << x << " "; cout << endl #define print_pair(a) #define FOR(i, fr, to) for (long long i = fr; i <= to; i++) #define RFOR(i, fr, to) for (long long i = fr; i >= to; i--) #define pb push_back #define eb emplace_back #define is insert #define F first #define S second using namespace std; using ll = long long; using ld = long double; using vll = vector<ll>; using vvll = vector<vll>; using vc = vector<char>; using vvc = vector<vc>; using pll = pair<ll, ll>; using vpll = vector<pll>; using vvpll = vector<vpll>; using stkll = stack<ll>; using qll = queue<ll>; using dqll = deque<ll>; using sll = set<ll>; using msll = multiset<ll>; using mll = map<ll, ll>; using vsll = vector<sll>; using sc = set<char>; using pcll = pair<char, ll>; ll dr[] = {0, 1, 0, -1}, dc[] = {-1, 0, 1, 0}; const ll INF = 1e18; ll MOD = 998244353; ll mod_add(ll u, ll v, ll mod = MOD) { u %= mod, v %= mod; return (u + v) % mod; } ll mod_sub(ll u, ll v, ll mod = MOD) { u %= mod, v %= mod; return (u - v + mod) % mod; } ll mod_mul(ll u, ll v, ll mod = MOD) { u %= mod, v %= mod; return (u * v) % mod; } ll mod_pow(ll u, ll p, ll mod = MOD) { u %= mod; if (p == 0) { return 1; } ll v = mod_pow(u, p / 2, mod) % mod; v = (v * v) % mod; return (p % 2 == 0) ? v : (u * v) % mod; } ll mod_inv(ll u, ll mod = MOD) { u %= mod; ll g = __gcd(u, mod); if (g != 1) { return -1; } return mod_pow(u, mod - 2, mod); } ll mod_div(ll u, ll v, ll mod = MOD) { u %= mod, v %= mod; return mod_mul(u, mod_inv(v), mod); } template<class T> vector<T> operator+(vector<T> a, vector<T> b) { a.insert(a.end(), b.begin(), b.end()); return a; } pll operator+(pll a, pll b) { pll ans = {a.first + b.first, a.second + b.second}; return ans; } template<class A, class B> ostream &operator<<(ostream &os, const pair<A, B> &p) { os << p.first << " " << p.second; return os; } template<class A, class B> istream &operator>>(istream &is, pair<A, B> &p) { is >> p.first >> p.second; return is; } template<class T> ostream &operator<<(ostream &os, const vector<T> &vec) { for (auto &x: vec) { os << x << " "; } return os; } template<class T> istream &operator>>(istream &is, vector<T> &vec) { for (auto &x: vec) { is >> x; } return is; } template<class T> ostream &operator<<(ostream &os, const set<T> &vec) { for (auto &x: vec) { os << x << " "; } return os; } template<class A, class B> ostream &operator<<(ostream &os, const map<A, B> d) { for (auto &x: d) { os << "(" << x.first << " " << x.second << ") "; } return os; } template<class A> void read_arr(A a[], ll from, ll to) { for (ll i = from; i <= to; i++) { cin >> a[i]; } } template<class A> void print_arr(A a[], ll from, ll to) { for (ll i = from; i <= to; i++) { cout << a[i] << " "; } cout << "\n"; } template<class A> void print_arr(A a[], ll n) { print_arr(a, 0, n - 1); } template<class A> void set_arr(A a[], A val, ll from, ll to) { for (ll i = from; i <= to; i++) { a[i] = val; } } template<class A> void set_arr(A a[], A val, ll n) { set_arr(a, val, 0, n - 1); } const ll N = 1e6 + 10; struct Node { ll t, cnt; vll vec, pre, suf; vll sort_pre, sort_suf; static Node base(ll t) { return Node{t, 0, vll(), vll(), vll(), vll(), vll()}; } friend Node operator+(const Node &a, const Node &b) { Node res; res.pre = a.pre; for (auto &x: b.vec) { res.pre.push_back(x + res.pre.back()); } res.suf = b.suf; reverse(all(res.suf)); for (ll i = sz(a.vec) - 1; i >= 0; i--) { res.suf.push_back(a.vec[i] + res.suf.back()); } reverse(all(res.suf)); res.sort_pre = res.pre, res.sort_suf = res.suf; sort(all(res.sort_pre)), sort(all(res.sort_suf)); res.cnt = a.cnt + b.cnt; for (auto &u: a.sort_suf) { ll x = a.t - u; res.cnt += upper_bound(all(b.sort_pre), x) - b.sort_pre.begin(); } res.vec = a.vec; res.vec.insert(res.vec.end(), b.vec.begin(), b.vec.end()); return res; } }; struct Smt { ll n; Node tree[4 * N]; const ll def= 0; void init(ll _n) { this->n = _n; } void build(ll a[N], ll id, ll l, ll r, ll t) { if (l == r) { vll cur = {a[l]}; tree[id] = {t, a[l] < t, cur, cur, cur, cur, cur}; cout << make_pair(l, r) << " " << tree[id].cnt << "\n"; return; } ll mid = (l + r) / 2; build(a, 2 * id, l, mid, t); build(a, 2 * id + 1, mid + 1, r, t); cout << make_pair(l, r) << " " << tree[id].cnt << "\n"; calc(id); } void build(ll a[N], ll t) { build(a, 1, 1, n, t); } void calc(ll id) { tree[id] = get_val(tree[2 * id], tree[2 * id + 1]); } static Node get_val(const Node& op1, const Node& op2) { return op1 + op2; } Node get(ll id, ll l, ll r, ll u, ll v, ll t) { if (l > v || r < u) { return Node::base(t); } if (l >= u && r <= v) { return tree[id]; } ll mid = (l + r) / 2; Node op1 = get(2 * id, l, mid, u, v, t); Node op2 = get(2 * id + 1, mid + 1, r, u ,v, t); return get_val(op1, op2); } Node get(ll u, ll v, ll t) { return get(1, 1, n, u, v, t); } }; ll n, m, k, q; ll a[N]; string s, s1, s2, s3; Smt smt; void init() { } ll get_ans() { } void single(ll t_id = 0) { ll t; cin >> n >> t; FOR(i, 1, n) { cin >> a[i]; } smt.init(n); smt.build(a, t); Node u = smt.get(1, n, t); cout << u.pre << "\n" << u.suf << "\n"; cout << u.cnt << "\n"; } void multiple() { ll t; cin >> t; for (ll i = 0; i < t; i++) { single(i); } } //#endif #if 0 void multiple() { while (cin >> n >> m) { if (n == 0 && m == 0) { break; } single(); } #endif int main(int argc, char *argv[]) { ios_base::sync_with_stdio(false); cin.tie(nullptr); init(); // freopen("feast.in", "r", stdin); // freopen("feast.out", "w", stdout); freopen("../input.txt", "r", stdin); multiple(); // single(); return 0; };
Editor is loading...