Untitled
unknown
plain_text
3 years ago
7.4 kB
10
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...