QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#497725 | #6533. Traveling in Cells | siunkyo | AC ✓ | 3168ms | 30352kb | C++20 | 4.6kb | 2024-07-29 16:28:38 | 2024-07-29 16:28:38 |
Judging History
answer
#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define ld long double
#define i64 __int128
#define PII pair<int,int>
#define endl '\n'
#define pb push_back
#define fi first
#define se second
#define YES cout << "YES" << endl
#define NO cout << "NO" << endl
//cout<<setiosflags(ios::fixed)<<setprecision(K);
const int dx[4] = {1,-1,0,0};
const int dy[4] = {0,0,1,-1};
const double eps = 1e-9;
const double PI = acos(-1.0);
const int INF = 1e9 + 7;
const int mod = 998244353;
const int N = 1e5 + 10,M = 1e5;
int n, q;
int v[N], c[N];
int tot; // T[i]:颜色为i的线段树的根
struct node{
int ls, rs, sum;
}tr[N * 200];
void update(int& p, int loc, int k, int l = 1, int r = M){ // 对于该颜色更新点loc的价值+1
if(!p) p = ++ tot;
tr[p].sum += k;
if(l == r) return ;
int mid = (l + r) >> 1;
if(loc <= mid) update(tr[p].ls, loc, k, l, mid);
else update(tr[p].rs, loc, k, mid + 1, r);
}
int query(int p, int ql, int qr, int l = 1, int r = M){ // 对于颜色i查询区间[ql, qr]该颜色的个数
if(!p) return 0;
if(ql <= l && r <= qr) return tr[p].sum;
int mid = (l + r) >> 1, ans = 0;
if(ql <= mid) ans += query(tr[p].ls, ql, qr, l, mid);
if(qr > mid) ans += query(tr[p].rs, ql, qr, mid + 1, r);
return ans;
}
template <typename T>
struct Fenwick {
int n;
vector<T> a;
Fenwick(int n = 0) {init(n);}
void init(int n) { this->n = n; a.assign(n, T());}
//在x处加上v
void add(int x, T v) {
for (int i = x + 1; i <= n; i += i & -i) {
a[i - 1] += v;
}
}
//取[0.x)的和
T sum(int x) {
auto ans = T();
for (int i = x; i > 0; i -= i & -i) {
ans += a[i - 1];
}
return ans;
}
//取[l,r)的和
T rangeSum(int l, int r) {
return sum(r) - sum(l);
}
//取已插入的元素从小到大排序后第k个元素的值,这里从0开始记录
int kth(T k) {
int x = 0;
for (int i = 1 << __lg(n); i; i /= 2) {
if (x + i <= n && k >= a[x + i - 1]) {
x += i;
k -= a[x - 1];
}
}
return x;
}
};
//Fenwick<int> fen(N);
//fen.add(x,y); //往x位置加上y
//fen.sum(x+1); //求[0,x+1)的和
void init(){
for(int i = 0;i <= tot;i ++) tr[i] = {0,0,0};
tot = 0;
}
void solve(){
cin >> n >> q;
init();
vector<int> root(n + 1);
for(int i = 1;i <= n;i ++){
cin >> c[i];
update(root[c[i]],i,1);
}
Fenwick<LL> fen(n + 1);
for(int i = 1;i <= n;i ++){
cin >> v[i];
fen.add(i,v[i]);
}
while(q --){
int op;
cin >> op;
if(op == 1){
int p,x;
cin >> p >> x;
update(root[c[p]],p,-1);
c[p] = x;
update(root[c[p]],p,1);
}else if(op == 2){
int p,x;
cin >> p >> x;
fen.add(p,-v[p]);
fen.add(p,x);
v[p] = x;
}else{
int x,k;
cin >> x >> k;
vector<int> a(k + 1);
for(int i = 1;i <= k;i ++) cin >> a[i];
auto check = [&](int ll){
LL sum = 0;
for(int i = 1;i <= k;i ++){
sum += query(root[a[i]],ll,x);
}
if(sum == x - ll + 1) return true;
return false;
};
auto checkk = [&](int rr){
LL sum = 0;
for(int i = 1;i <= k;i ++){
sum += query(root[a[i]],x,rr);
}
if(sum == rr - x + 1) return true;
return false;
};
int l = 1,r = x,ansl = -1,ansr = -1;
while(l <= r){
int mid = (l + r) / 2;
if(check(mid) == true){
r = mid - 1;
ansl = mid;
}else{
l = mid + 1;
}
}
l = x,r = n;
while(l <= r){
int mid = (l + r) / 2;
if(checkk(mid) == true){
l = mid + 1;
ansr = mid;
}else{
r = mid - 1;
}
}
// cout << ansl << ' ' << ansr << endl;
cout << fen.rangeSum(ansl,ansr + 1) << endl;
}
}
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int T = 1;
cin >> T;
while(T --){
solve();
}
return 0;
}
详细
Test #1:
score: 100
Accepted
time: 0ms
memory: 3536kb
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: 353ms
memory: 3632kb
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: 665ms
memory: 3572kb
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: 1679ms
memory: 6096kb
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: 0
Accepted
time: 1637ms
memory: 11788kb
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...
result:
ok 199749 numbers
Test #6:
score: 0
Accepted
time: 2368ms
memory: 21100kb
input:
3 100000 100000 173 276 418 362 183 321 401 316 193 426 212 126 206 409 382 443 405 412 259 233 356 355 340 41 354 447 421 464 436 436 329 239 427 415 452 424 174 294 220 413 293 456 140 304 438 462 418 345 160 296 443 234 455 452 396 347 438 413 235 416 363 186 340 285 340 457 392 359 451 310 431 1...
output:
832547880 1825993219 676042867 310750190 650714631 657481975 1279322 838513014 453432678 940357183 846050641 631145680 278723792 689448062 154699248 45678908 56518237 839298643 611124630 499104412 324172054 742064269 626600147 728123335 602272914 45485542 868574266 876207167 342300121 917221167 7055...
result:
ok 200119 numbers
Test #7:
score: 0
Accepted
time: 1038ms
memory: 29088kb
input:
3 100000 100000 66046 49249 42478 61684 59308 38366 66208 38769 50465 63701 50193 47811 50312 56793 58616 63383 58390 17546 23446 57532 59030 63009 62771 46338 52747 54677 68893 58360 56617 42330 65075 51193 65417 67035 54247 54323 39404 61892 42821 30094 67958 46206 53273 28507 65864 42364 64063 46...
output:
787181803 340358691 794440759 970166774 501256821 822531703 505349238 299646179 317091968 718901636 589846615 490283989 1183290 98835675 715091970 941598117 81757251 249078591 559980949 587721512 408541247 542135558 217103011 916103998 422219165 786665928 215595722 309683697 275650079 660176857 9707...
result:
ok 199750 numbers
Test #8:
score: 0
Accepted
time: 1339ms
memory: 30352kb
input:
3 100000 100000 2 2 3 1 1 3 1 3 1 1 1 1 1 3 3 2 3 1 1 3 3 3 2 1 2 3 2 3 1 1 2 2 2 3 1 1 3 3 3 3 1 3 1 1 3 2 3 1 1 2 3 3 2 1 3 1 1 1 2 3 3 2 3 3 1 1 2 2 2 1 1 3 1 1 2 2 1 3 3 3 2 1 1 2 1 2 3 3 3 2 2 3 3 2 3 2 1 3 3 3 2 2 3 1 3 2 3 2 3 1 2 2 1 2 1 3 3 3 2 2 3 1 3 2 1 1 1 2 1 2 2 1 3 1 2 1 1 3 3 2 3 3 ...
output:
50035938417434 50036115992265 50036315519498 3975347985 50036470174447 50036191297549 7222739016 50036248757817 1886106914 50037215621946 50037215621946 50037215621946 50037789462602 50037789462602 50038171558883 50038171558883 50038171558883 50037356071541 50037356071541 840665598 50037454922987 19...
result:
ok 144306 numbers
Test #9:
score: 0
Accepted
time: 1614ms
memory: 27984kb
input:
3 100000 100000 60522 14575 36426 79445 48772 90081 33447 90629 3497 47202 7775 94325 63982 4784 68417 2156 31932 35902 95728 78537 23857 30739 86918 29211 39679 38506 63340 86568 61868 60016 87940 96263 24593 1449 36991 90310 23355 77068 11431 8580 91757 49218 74934 94328 63676 29355 96221 99080 95...
output:
49028798194951 49028798194951 49028798194951 445374817 49028798194951 49028798194951 49028798194951 49028798194951 49029113499144 49029113499144 49029113499144 568084096 49029252709603 49029252709603 49029252709603 49029252709603 49029252709603 49029252709603 49029252709603 142834402 644706458 49029...
result:
ok 151536 numbers
Test #10:
score: 0
Accepted
time: 3168ms
memory: 10412kb
input:
3 100000 100000 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 3 1 2 ...
output:
50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 50042766706436 ...
result:
ok 300000 numbers