QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#726077 | #6533. Traveling in Cells | He_ng | TL | 4699ms | 515232kb | C++20 | 4.9kb | 2024-11-08 21:31:32 | 2024-11-08 21:31:33 |
Judging History
answer
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pii pair<int, int>
const int N = 1e5 + 10;
const int M = 2e3;
const int MAXA = 1e5 + 5;
#define endl '\n'
struct DynamicSegtree {
struct node {
int data, tag;
node *lc, *rc;
node(int data_ = 0) : data(data_), tag(0), lc(nullptr), rc(nullptr) {}
void up() {
data = 0;
if (lc != nullptr)
data += lc->data;
if (rc != nullptr)
data += rc->data;
return;
}
void down(int l, int r) {
if (lc == nullptr)
lc = new node;
if (rc == nullptr)
rc = new node;
int mid = (l + r) >> 1;
lc->data += (mid - l + 1) * tag;
lc->tag += tag;
rc->data += (r - mid) * tag;
rc->tag += tag;
tag = 0;
}
} *rt = new node(0);
void modify(node *rt, int L, int R, int l, int r, int val) {
if (l <= L && R <= r) {
rt->data += (R - L + 1) * val;
rt->tag += val;
return;
}
rt->down(L, R);
int mid = (L + R) >> 1;
if (l <= mid)
modify(rt->lc, L, mid, l, r, val);
if (r > mid)
modify(rt->rc, mid + 1, R, l, r, val);
rt->up();
return;
}
int query(node *rt, int L, int R, int l, int r) {
if (l <= L && R <= r) {
return rt->data;
}
rt->down(L, R);
int mid = (L + R) >> 1;
int ret = 0;
if (l <= mid)
ret += query(rt->lc, L, mid, l, r);
if (r > mid)
ret += query(rt->rc, mid + 1, R, l, r);
return ret;
}
};
int n, q;
int val[N];
int lowbit(int x) { return x & -x; }
void update1(int r, int k) {
for (int i = r; i <= n; i += lowbit(i))
val[i] += k;
}
int get_sum(int l, int r) {
int ans = 0;
for (int i = r; i; i -= lowbit(i))
ans += val[i];
for (int i = l - 1; i; i -= lowbit(i))
ans -= val[i];
return ans;
}
void solve() {
cin >> n >> q;
vector<DynamicSegtree> tree(n + 1);
vector<int> c(n + 1);
vector<int> v(n + 1);
for (int i = 1; i <= n; i++) {
cin >> c[i];
tree[c[i]].modify(tree[c[i]].rt, 1, MAXA,i,i, 1);
val[i] = 0;
}
for (int i = 1; i <= n; i++) {
cin >> v[i];
update1(i, v[i]);
}
while (q--) {
int x;
cin >> x;
if (x == 1) {
int p, y;
cin >> p >> y;
tree[c[p]].modify(tree[c[p]].rt, 1, MAXA, p,p, -1);
tree[y].modify(tree[y].rt, 1, MAXA,p, p, 1);
c[p] = y;
} else if (x == 2) {
int p, y;
cin >> p >> y;
update1(p, y - v[p]);
v[p] = y;
} else if (x == 3) {
int st, k;
cin >> st >> k;
vector<int> a(k + 1);
for (int i = 1; i <= k; i++) {
cin >> a[i];
}
sort(a.begin(), a.end());
a.erase(unique(a.begin()+1, a.end()), a.end());
k = a.size()-1;
int l = 1;
int r = st;
auto check1 = [&](int mid, const vector<int> &a, int st) -> bool {
int sum = 0;
for (int i = 1; i <= k; i++) {
int cnt = tree[a[i]].query(tree[a[i]].rt, 1, MAXA, mid, st);
sum += cnt;
}
return sum == st - mid + 1;
};
auto check2 = [&](int mid, const vector<int> &a, int st) -> bool {
int sum = 0;
for (int i = 1; i <= k; i++) {
int cnt = tree[a[i]].query(tree[a[i]].rt, 1, MAXA, st, mid);
sum += cnt;
}
return sum == mid - st + 1;
};
while (l <= r) {
int mid = (l + r) >> 1;
if (check1(mid, a, st))
r = mid-1;
else
l = mid + 1;
}
int ansl = l;
l = st;
r = n;
while (l <= r) {
int mid = (l + r + 1) >> 1;
if (check2(mid, a, st))
l = mid+1;
else
r = mid - 1;
}
int ansr = r;
cout << get_sum(ansl, ansr) << endl;
}
}
return;
}
signed main() {
ios::sync_with_stdio(false);
cin.tie(0);
int t = 1;
cin >> t;
while (t--)
solve();
return 0;
}
/*
1
5 10
1 2 3 1 2
1 10 100 1000 10000
3 3 1 3
3 3 2 2 3
2 5 20000
2 3 200
3 3 2 1 3
3 3 3 1 2 3
1 3 4
2 1 100000
1 2 2
3 1 2 1 2
*/
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 3636kb
input:
2 5 10 1 2 3 1 2 1 10 100 1000 10000 3 3 1 3 3 3 2 2 3 2 5 20000 2 3 200 3 3 2 1 3 3 3 3 1 2 3 1 3 4 2 1 100000 1 2 2 3 1 2 1 2 4 1 1 2 3 4 1000000 1000000 1000000 1000000 3 4 4 1 2 3 4
output:
100 110 1200 21211 100010 4000000
result:
ok 6 numbers
Test #2:
score: 0
Accepted
time: 743ms
memory: 515232kb
input:
20000 15 15 1 1 3 3 2 3 3 3 3 3 2 3 2 3 3 634593973 158136379 707704004 998369647 831633445 263092797 937841412 451774682 552617416 483763379 50360475 794662797 74247465 537217773 901809831 3 6 4 1 3 5 10 3 5 7 1 2 3 4 5 9 10 3 4 3 3 8 9 2 13 608033129 3 15 2 3 5 1 9 3 3 8 4 1 3 7 10 2 6 727472865 3...
output:
2689089686 8377825475 1706073651 1439027604 2689089686 792730352 8904867081 8904867081 8270273108 831633445 692051585 2782432626 697783016 883944422 184057757 287523250 184057757 696388752 184057757 1675459344 2667693025 2614711258 4006992373 1767091974 5348541057 5348541057 390631780 2290216252 942...
result:
ok 200062 numbers
Test #3:
score: 0
Accepted
time: 1260ms
memory: 312880kb
input:
2000 150 150 8 3 8 8 8 6 8 4 2 7 6 8 8 5 8 7 7 8 5 6 8 8 6 8 8 8 8 7 8 6 6 8 8 8 6 2 3 4 8 8 7 8 5 8 2 6 8 7 8 8 6 8 6 8 3 8 8 8 8 4 7 8 7 3 7 6 7 5 5 8 6 8 8 6 3 8 6 7 6 8 8 7 4 8 6 7 8 7 7 7 7 8 8 8 8 2 5 2 8 8 6 7 6 3 8 8 7 8 8 8 6 6 8 6 6 7 5 8 8 8 7 8 7 7 6 8 8 8 8 8 8 6 5 7 5 5 8 7 8 7 7 7 6 5...
output:
4449391171 3290849667 852793841 5178673994 995994209 11431868919 4327723427 5071541023 3032743466 962345334 2997656427 4534278452 3851900075 3611231417 5071541023 1477584218 1299005818 1299005818 2145605244 854143763 886347565 2081234124 2333808475 2455955801 4179722063 2328504333 1473735464 4107685...
result:
ok 199987 numbers
Test #4:
score: 0
Accepted
time: 4699ms
memory: 296480kb
input:
10 30000 30000 3 4 2 4 4 4 4 3 4 3 4 3 4 3 4 4 2 4 4 4 4 4 3 3 3 4 3 4 3 4 3 3 4 2 4 3 3 3 3 4 3 4 4 4 4 2 3 3 4 2 3 4 4 4 4 1 4 4 4 4 4 4 4 4 3 3 3 4 4 4 4 4 2 3 4 4 4 4 3 4 4 3 3 3 4 4 3 4 4 2 3 4 4 4 4 3 2 4 3 4 3 2 4 4 3 4 2 2 4 4 4 4 2 4 3 2 4 4 3 4 4 4 2 4 4 3 2 3 2 3 3 3 4 2 4 3 4 1 4 3 4 4 4...
output:
6959437173 934970676 72461245502 8365928740 8384151048 984567228 12482909122 1904927816 15134139942 3759040688 92670874909 332468911 5936663371 3562978848 1300592004 10314009201 5581540905 131246926443 15087184135864 4077066271 1124704817 1520626740 4388174158 744377942 2770411457 6231852240 1508724...
result:
ok 200135 numbers
Test #5:
score: -100
Time Limit Exceeded
input:
3 100000 100000 6 6 2 6 5 3 6 5 4 6 4 6 6 6 6 5 2 5 2 6 6 6 1 6 5 6 4 5 6 6 5 4 5 4 3 4 5 5 6 6 5 6 6 5 2 5 6 5 4 2 5 6 6 6 5 2 5 6 6 4 5 6 3 3 6 5 6 5 5 5 5 4 4 4 4 3 6 5 4 5 6 5 6 6 6 6 3 6 5 6 5 4 3 5 6 4 5 3 6 5 3 5 6 4 6 5 4 5 5 5 2 5 4 6 6 3 5 5 5 5 5 4 5 5 6 5 5 6 6 6 5 5 4 6 5 4 4 2 6 6 6 5 ...
output:
753014823 938372065 5655899645 168297301 14372254310 1066586326 3520855082 2591792266 2321844837 64378192092 250581310 1845085639 1402247975 198007248 2157074263 2743531397 3433471688 10332600792 1085086491 4845484125 50019185900042 4036199358 154762798 50019185900042 1221387905 11240790307 10537215...