이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "bits/stdc++.h"
#include <stdio.h>
#include <stdlib.h>
#include <assert.h>
using namespace std;
const int MOD = 1e9 + 7;
mt19937_64 rng((int)std::chrono::steady_clock::now().time_since_epoch().count());
int rnd(int x, int y) {
int u = uniform_int_distribution<int>(x, y)(rng); return u;
}
#define inbuf_len 1 << 16
#define outbuf_len 1 << 16
int GetBestPosition(int N, int C, int R, int *K, int *S, int *E);
#include "bits/stdc++.h"
using namespace std;
const int MAXN = 1e5 + 10;
int st[17][MAXN];
int query(int l, int r) {
int k = 32 - __builtin_clz(r-l+1) - 1; return max(st[k][l], st[k][r - (1<<k) + 1]);
}
// WARNING: This template lacks testing for query 5, 6, 7
// WARNING: This template is over 700 lines long
// NOTE: 0 is allowed as index in this segtree version
struct segtree_basic {
//Segtree that supports all basic operations
//Updates: Range :=, Range +=
//Basic queries: Range min, Range max, Range sum
//Advanced queries: Range first >= v, Range first <= v, Range last >= v, Range last <= v
//https://codeforces.com/edu/course/2/lesson/5/2/practice Problem A, D, E, F
//https://codeforces.com/edu/course/2/lesson/5/3/practice Problem C
//https://codeforces.com/edu/course/2/lesson/5/4/practice Problem A
//#define int long long please
struct node {
int upd = 0;
char type = '?';
int mi = 0, ma = 0, sum = 0;
};
int stok;
vector<node> st;
void u1(int l, int r, int constl, int constr, int idx, int val) { // Range :=
if(l <= constl && constr <= r) {
st[idx].upd = val;
st[idx].type = ':';
st[idx].mi = val;
st[idx].ma = val;
st[idx].sum = val * (constr - constl + 1);
return;
}
int mid = (constl + constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(mid<l || r<constl) u1(l, r, mid + 1, constr, 2*idx+2, val);
else if(constr<l || r<mid+1) u1(l, r, constl, mid, 2*idx+1, val);
else {
u1(l, r, constl, mid, 2*idx+1, val); u1(l, r, mid + 1, constr, 2*idx+2, val);
}
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
void u2(int l, int r, int constl, int constr, int idx, int val) { // Range +=
if(l <= constl && constr <= r) {
st[idx].upd += val;
if(st[idx].type == '?') st[idx].type = '+';
st[idx].mi += val;
st[idx].ma += val;
st[idx].sum += val * (constr - constl + 1);
return;
}
int mid = (constl + constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(mid<l || r<constl) u2(l, r, mid + 1, constr, 2*idx+2, val);
else if(constr<l || r<mid+1) u2(l, r, constl, mid, 2*idx+1, val);
else {
u2(l, r, constl, mid, 2*idx+1, val); u2(l, r, mid + 1, constr, 2*idx+2, val);
}
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
int qu1(int l, int r, int constl, int constr, int idx) { // Range min
if(l<=constl && constr<=r) return st[idx].mi;
int mid = (constl + constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(mid<l || r<constl) return qu1(l, r, mid+1, constr, 2*idx+2);
else if(constr<l || r<mid+1) return qu1(l, r, constl, mid, 2*idx+1);
return min(qu1(l, r, constl, mid, 2*idx+1), qu1(l, r, mid+1, constr, 2*idx+2));
}
int qu2(int l, int r, int constl, int constr, int idx) { // Range max
if(l<=constl && constr<=r) return st[idx].ma;
int mid = (constl + constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(mid<l || r<constl) return qu2(l, r, mid+1, constr, 2*idx+2);
else if(constr<l || r<mid+1) return qu2(l, r, constl, mid, 2*idx+1);
return max(qu2(l, r, constl, mid, 2*idx+1), qu2(l, r, mid+1, constr, 2*idx+2));
}
int qu3(int l, int r, int constl, int constr, int idx) { // Range sum
if(l<=constl && constr<=r) return st[idx].sum;
int mid = (constl + constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(mid<l || r<constl) return qu3(l, r, mid + 1, constr, 2*idx + 2);
else if(constr < l || r <mid+ 1) return qu3(l, r, constl, mid, 2*idx + 1);
else {
return qu3(l, r, constl, mid, 2*idx + 1) + qu3(l, r, mid + 1, constr, 2*idx + 2);
}
}
int qu4(int l, int r, int constl, int constr, int idx, int val) { // First element ≥ v
if(l<=constl && constr<=r) {
if(st[idx].ma < val) return -1;
while(constl < constr) {
int mid = (constl + constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(st[2*idx+1].ma >= val) idx = 2*idx + 1, constr = mid;
else idx = 2*idx+2, constl = mid + 1;
}
return constl;
}
int mid = (constl+constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(mid<l || r<constl) return qu4(l, r, mid+1, constr, 2*idx+2, val);
else if(constr<l || r<mid+1) return qu4(l, r, constl, mid, 2*idx+1, val);
else {
int oho = qu4(l, r, constl, mid, 2*idx+1, val);
if(oho != -1) return oho;
return qu4(l, r, mid+1, constr, 2*idx+2, val);
}
}
int qu5(int l, int r, int constl, int constr, int idx, int val) { // First element ≤ v
if(l<=constl && constr<=r) {
if(st[idx].mi > val) return -1;
while(constl < constr) {
int mid = (constl + constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(st[2*idx+1].mi <= val) idx = 2*idx + 1, constr = mid;
else idx = 2*idx+2, constl = mid + 1;
}
return constl;
}
int mid = (constl+constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(mid<l || r<constl) return qu5(l, r, mid+1, constr, 2*idx+2, val);
else if(constr<l || r<mid+1) return qu5(l, r, constl, mid, 2*idx+1, val);
else {
int oho = qu5(l, r, constl, mid, 2*idx+1, val);
if(oho != -1) return oho;
return qu5(l, r, mid+1, constr, 2*idx+2, val);
}
}
int qu6(int l, int r, int constl, int constr, int idx, int val) { // Last element ≥ v
if(l<=constl && constr<=r) {
if(st[idx].ma < val) return -1;
while(constl < constr) {
int mid = (constl + constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(st[2*idx+2].ma >= val) idx = 2*idx + 2, constl = mid + 1;
else idx = 2*idx + 1, constr = mid;
}
return constl;
}
int mid = (constl+constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(mid<l || r<constl) return qu6(l, r, mid+1, constr, 2*idx+2, val);
else if(constr<l || r<mid+1) return qu6(l, r, constl, mid, 2*idx+1, val);
else {
int oho = qu6(l, r, mid+1, constr, 2*idx+2, val);
if(oho != -1) return oho;
return qu6(l, r, constl, mid, 2*idx+1, val);
}
}
int qu7(int l, int r, int constl, int constr, int idx, int val) { // Last element ≤ v
if(l<=constl && constr<=r) {
if(st[idx].mi > val) return -1;
while(constl < constr) {
int mid = (constl + constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(st[2*idx+2].mi <= val) idx = 2*idx + 2, constl = mid + 1;
else idx = 2*idx + 1, constr = mid;
}
return constl;
}
int mid = (constl+constr) >> 1;
if(st[idx].type != '?') {
if(st[idx].type == ':') {
st[2*idx+1].upd = st[idx].upd;
st[2*idx+1].type = ':';
st[2*idx+1].mi = st[idx].upd;
st[2*idx+1].ma = st[idx].upd;
st[2*idx+1].sum = st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd = st[idx].upd;
st[2*idx+2].type = ':';
st[2*idx+2].mi = st[idx].upd;
st[2*idx+2].ma = st[idx].upd;
st[2*idx+2].sum = st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
else {
st[2*idx+1].upd += st[idx].upd;
if(st[2*idx+1].type == '?') st[2*idx+1].type = '+';
st[2*idx+1].mi += st[idx].upd;
st[2*idx+1].ma += st[idx].upd;
st[2*idx+1].sum += st[idx].upd * (mid - constl + 1);
st[2*idx+2].upd += st[idx].upd;
if(st[2*idx+2].type == '?') st[2*idx+2].type = '+';
st[2*idx+2].mi += st[idx].upd;
st[2*idx+2].ma += st[idx].upd;
st[2*idx+2].sum += st[idx].upd * (constr - mid);
st[idx].upd = 0;
st[idx].type = '?';
st[idx].mi = min(st[2*idx+1].mi, st[2*idx+2].mi);
st[idx].ma = max(st[2*idx+1].ma, st[2*idx+2].ma);
st[idx].sum = st[2*idx+1].sum + st[2*idx+2].sum;
}
}
if(mid<l || r<constl) return qu7(l, r, mid+1, constr, 2*idx+2, val);
else if(constr<l || r<mid+1) return qu7(l, r, constl, mid, 2*idx+1, val);
else {
int oho = qu7(l, r, mid+1, constr, 2*idx+2, val);
if(oho != -1) return oho;
return qu7(l, r, constl, mid, 2*idx+1, val);
}
}
public:
void resize(int k) {
stok = k;
st.resize(4 * k + 10);
}
void range_assign(int l, int r, int v) {
u1(l, r, 0, stok, 0, v);
}
void range_add(int l, int r, int v) {
u2(l, r, 0, stok, 0, v);
}
int query_min(int l, int r) {
return qu1(l, r, 0, stok, 0);
}
int query_max(int l, int r) {
return qu2(l, r, 0, stok, 0);
}
int query_sum(int l, int r) {
return qu3(l, r, 0, stok, 0);
}
int query_firstAtLeast(int l, int r, int v) {
return qu4(l, r, 0, stok, 0, v);
}
int query_firstAtMost(int l, int r, int v) {
return qu5(l, r, 0, stok, 0, v);
}
int query_lastAtLeast(int l, int r, int v) {
return qu6(l, r, 0, stok, 0, v);
}
int query_lastAtMost(int l, int r, int v) {
return qu7(l, r, 0, stok, 0, v);
}
};
int GetBestPosition(int N, int C, int R, int *K, int *S, int *E) {
segtree_basic dsu;
dsu.resize(N);
for(int i=0; i<N; i++) dsu.range_assign(i, i, i);
for(int i=0; i<C; i++) {
int l = dsu.query_firstAtLeast(0, N - 1, S[i]);
int r = dsu.query_lastAtMost(0, N - 1, E[i]);
if(l == -1) l = 0;
if(r == -1) r = N - 1;
if(l < r) dsu.range_assign(l + 1, r, dsu.query_min(l, l));
if(r + 1 <= N - 1) dsu.range_add(r + 1, N - 1, -(E[i] - S[i]));
S[i] = l, E[i] = r;
}
/*
for(int i=0; i<C; i++) {
cout << "Range #" << i <<": " << S[i] << " " << E[i] << "\n";
}
*/
int opt_cnt = 0, opt_id = 0;
int d[N + 1];
for(int i=0; i<N + 1; i++) d[i] = 0;
for(int i=0; i<N-1; i++) st[0][i] = K[i];
for(int i=1; i<17; i++) {
for(int j=0; j<N-1 && j + (1<<i) - 1 < N - 1; j++) {
st[i][j] = max(st[i-1][j], st[i-1][j+(1<<(i-1))]);
}
}
for(int i=0; i<C; i++) {
if(query(S[i], E[i] - 1) < R) {
d[S[i]]++;
d[E[i] + 1]--;
}
}
int cur = 0;
for(int i=0; i<N; i++) {
cur += d[i];
if(cur > opt_cnt) {
opt_cnt = cur, opt_id = i;
}
}
return opt_id;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |