QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#320798 | #3850. DJ Darko | algotester# | WA | 242ms | 66020kb | C++14 | 6.3kb | 2024-02-03 21:32:02 | 2024-02-03 21:32:03 |
Judging History
answer
#include "bits/stdc++.h"
using namespace std;
#define FOR(i,a,b) for (LL i = (a); i < (b); i++)
#define RFOR(i,b,a) for (LL i = (b) - 1; i >= (a); i--)
#define ITER(it,a) for (__typeof(a.begin()) it = a.begin(); it != a.end(); it++)
#define FILL(a,value) memset(a, value, sizeof(a))
#define SZ(a) (LL)a.size()
#define ALL(a) a.begin(), a.end()
#define PB push_back
#define MP make_pair
typedef long long LL;
typedef vector<LL> VI;
typedef pair<LL, LL> PII;
const double PI = acos(-1.0);
const LL INF = 1000 * 1000 * 1000 + 7;
const LL LINF = INF * (LL) INF;
const LL MAX = 1000200;
// const LL MAX = 100;
LL A[MAX];
LL B[MAX];
LL SB[MAX];
LL mrand()
{
return abs((rand() << 16) ^ rand());
}
struct node
{
LL l, r; // children
LL y; // priority (should be random and different)
LL cnt; // size of subtree
LL par; // parent of the vertex
LL val; // value of the vertex
LL push; // reverse push
LL ll;
LL rr;
void init(LL val, LL ll, LL rr) // init with value
{
l = r = -1;
y = mrand();
cnt = 1;
par = -1;
this->val = val;
this->ll = ll;
this->rr = rr;
this->push = 0;
}
};
// Minimum on subtree + reverse
struct Treap
{
node A[MAX];
LL sz;
LL getCnt(LL x)
{
if (x == -1) return 0;
return A[x].cnt;
}
LL newNode(LL val, LL ll, LL rr)
{
A[sz].init(val, ll, rr);
sz++;
return sz - 1;
}
LL PB(LL root, LL val, LL ll, LL rr)
{
return merge(root, newNode(val, ll, rr));
}
void upd(LL x)
{
if (x == -1) return;
A[x].cnt = getCnt(A[x].l) + getCnt(A[x].r) + 1;
}
void add(LL x, LL val)
{
if (x == -1) return;
A[x].val += val;
A[x].push += val;
}
void push(LL x)
{
if (x == -1 || A[x].push == 0) return;
add(A[x].l, A[x].push);
add(A[x].r, A[x].push);
A[x].push = 0;
}
PII split(LL x, LL cnt)
{
if (x == -1) return MP(-1, -1);
if (cnt == 0) return MP(-1, x);
push(x);
LL left = getCnt(A[x].l);
PII res;
if (left >= cnt)
{
if (A[x].l != -1) A[A[x].l].par = -1;
res = split(A[x].l, cnt);
A[x].l = res.second;
if (res.second != -1) A[res.second].par = x;
res.second = x;
}
else
{
if (A[x].r != -1) A[A[x].r].par = -1;
res = split(A[x].r, cnt - left - 1);
A[x].r = res.first;
if (res.first != -1) A[res.first].par = x;
res.first = x;
}
upd(x);
return res;
}
LL merge(LL x, LL y)
{
if (x == -1) return y;
if (y == -1) return x;
LL res;
//if (mrand() % (getCnt(x) + getCnt(y)) < getCnt(x))
if (A[x].y > A[y].y)
{
push(x);
if (A[x].r != -1) A[A[x].r].par = -1;
res = merge(A[x].r, y);
A[x].r = res;
if (res != -1) A[res].par = x;
res = x;
}
else
{
push(y);
if (A[y].l != -1) A[A[y].l].par = -1;
res = merge(x, A[y].l);
A[y].l = res;
if (res != -1) A[res].par = y;
res = y;
}
upd(res);
return res;
}
LL getCntLeft(LL x)
{
LL prev = -1;
LL res = 0;
while(true)
{
if (prev != A[x].l) res += getCnt(A[x].l);
LL par = A[x].par;
if (par == -1) break;
prev = x;
if (A[par].r == x) res++;
x = par;
}
return res;
}
LL dfs_find(LL x, LL pos)
{
// cout<<x<<' '<<pos<<": "<<A[x].ll<<' '<<A[x].rr<<' '<<A[x].l<<' '<<A[x].r<<endl;
if (A[x].ll <= pos && pos <= A[x].rr) return x;
if (pos < A[x].ll) return dfs_find(A[x].l, pos);
return dfs_find(A[x].r, pos);
}
void get_v(LL x, vector<PII>& v)
{
if (x == -1) return;
push(x);
get_v(A[x].l, v);
LL val = A[x].val;
// LL cnt = A[x].rr - A[x].ll + 1;
LL sum = SB[A[x].rr];
if (A[x].ll != 0) sum -= SB[A[x].ll-1];
v.PB(MP(val, sum));
get_v(A[x].r, v);
}
void prLL(LL x)
{
if (x == -1) return;
push(x);
prLL(A[x].l);
cout<<A[x].val<<" ("<<A[x].ll<<' '<<A[x].rr<<"), ";
prLL(A[x].r);
}
} T;
LL split_vertex_left(LL x, LL mid)
{
LL y = T.newNode(T.A[x].val, T.A[x].ll, mid - 1);
T.A[x].ll = mid;
// cout<<"@@ "<<x<<": "<<T.A[x].ll<<' '<<T.A[x].rr<<endl;
// cout<<"@@ "<<y<<": "<<T.A[y].ll<<' '<<T.A[y].rr<<endl;
return y;
}
LL split_vertex_right(LL x, LL mid)
{
// cout<<"# "<<x<<' '<<T.A[x].ll<<' '<<T.A[x].rr<<": "<<mid<<endl;
LL y = T.newNode(T.A[x].val, mid + 1, T.A[x].rr);
T.A[x].rr = mid;
// cout<<"## "<<x<<": "<<T.A[x].ll<<' '<<T.A[x].rr<<endl;
// cout<<"## "<<y<<": "<<T.A[y].ll<<' '<<T.A[y].rr<<endl;
return y;
}
pair<LL, PII> getLR(LL root, LL l, LL r)
{
LL x = T.dfs_find(root, l);
LL y = T.dfs_find(root, r);
// cout<<root<<' '<<l<<' '<<r<<endl;
// cout<<"! "<<x<<' '<<y<<endl;
PII a = T.split(root, T.getCntLeft(x));
PII b = T.split(a.second, T.getCntLeft(y) + 1);
root = b.first;
if (T.A[x].ll != l)
{
LL v = split_vertex_left(x, l);
a.first = T.merge(a.first, v);
}
if (T.A[y].rr != r)
{
LL v = split_vertex_right(y, r);
b.second = T.merge(v, b.second);
}
return MP(a.first, MP(b.first, b.second));
}
LL root;
void update(LL l, LL r, LL x)
{
auto R = getLR(root, l, r);
LL mr = R.second.first;
T.add(mr, x);
root = T.merge(R.first, T.merge(R.second.first, R.second.second));
}
LL update(LL l, LL r)
{
// cout<<"update "<<l<<' '<<r<<endl;
auto R = getLR(root, l, r);
// cout<<R.first<<' '<<R.second.first<<' '<<R.second.second<<endl;
vector<PII> v;
T.get_v(R.second.first, v);
// FOR (i, 0, SZ(v))
// {
// cout<<v[i].first<<' '<<v[i].second<<", ";
// }
// cout<<endl;
sort(ALL(v));
LL s = 0;
FOR (i, 0, SZ(v))
{
s += v[i].second;
}
LL c = 0;
LL val = -1;
FOR (i, 0, SZ(v))
{
c += v[i].second;
if (c * 2 >= s)
{
val = v[i].first;
break;
}
}
LL x = T.newNode(val, l, r);
root = T.merge(R.first, T.merge(x, R.second.second));
return val;
}
int main(int argc, char* argv[])
{
//ios::sync_with_stdio(false); cin.tie(0);
// cout<<"&"<<endl;
int n, q;
scanf("%d%d", &n, &q);
FOR (i, 0, n)
{
int x;
scanf("%d", &x);
A[i] = x;
}
FOR (i, 0, n)
{
int x;
scanf("%d", &x);
B[i] = x;
}
FOR (i, 0, n)
{
SB[i] = B[i];
if (i != 0) SB[i] += SB[i-1];
}
// cout<<"*"<<endl;
root = -1;
FOR (i, 0, n)
{
root = T.PB(root, A[i], i, i);
}
FOR (i, 0, q)
{
// cout<<"===> ";
// T.prLL(root);
// cout<<endl;
// cout<<"! "<<i<<endl;
int t;
scanf("%d", &t);
if (t == 1)
{
int l, r, x;
scanf("%d%d%d", &l, &r, &x);
l--;
r--;
update(l, r, x);
}
if (t == 2)
{
int l, r;
scanf("%d%d", &l, &r);
l--;
r--;
LL res = update(l, r);
printf("%lld\n", res);
}
}
// cout<<"===> ";
// T.prLL(root);
// cout<<endl;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Wrong Answer
time: 242ms
memory: 66020kb
input:
200000 200000 185413631 745038744 881479208 394948467 101727403 796960399 284541402 80768155 286582974 546327406 197495962 552359542 727479505 437895671 143092948 7626834 741609268 540494577 298656274 548860413 41137417 210529949 658779847 161355446 486548926 602900245 119414972 310187428 238177860 ...
output:
462406736 1749348519 1749348519 467651597 874061694 874061694 874061694 874061694 -1521749 874061694 -893932302 -425504259 -425504259 332034115 332034115 86195778 86195778 86195778 86195778 86195778 -425504259 -165776398 -165776398 -165776398 -916781043 -254293799 -254293799 -254293799 -254293799 -8...
result:
wrong answer 93rd lines differ - expected: '-2052051040', found: '-1402522055'