QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#525414 | #1823. Permutation CFG | RngBased# | AC ✓ | 175ms | 21072kb | C++17 | 4.9kb | 2024-08-20 16:16:10 | 2024-08-20 16:16:12 |
Judging History
answer
#include <bits/stdc++.h>
#define ll long long
#define pii pair<int, int>
#define pll pair<ll, ll>
#define F first
#define S second
#define all(x) x.begin(), x.end()
using namespace std;
const ll INF = 1e13;
int N, S, Q, p[100005], pos[100005], idx[200005];
ll ans[200005];
ll C(ll n, ll k)
{
if (k > n) return 0;
ll a = 1;
for (int i = 0; i < k; i++)
a = min(a * (n - i), INF);
for (int i = 1; i <= k; i++)
a /= i;
return a;
}
struct BIT
{
ll bit[100005];
void init()
{
fill(bit, bit + N + 1, 0);
}
void modify(int p, ll v)
{
for (; p <= N; p += (p & -p))
bit[p] += v;
}
ll query(int p)
{
ll a = 0;
for (; p; p -= (p & -p))
a += bit[p];
return a;
}
ll bsearch(ll v)
{
ll r = 0;
for (int p = (1 << __lg(N)); p >= 1; p >>= 1)
if (r + p <= N && bit[r + p] < v)
v -= bit[r + p], r += p;
return r;
}
} blen, bcnt, bacc;
/*
1 1 1 1 1
5 4 3 2 1
15 10 6 3 1
35 20 10 4 1
70 35 15 5 1
126 56 21 6 1
// s = 1: count number
// s = 2: count
// s >= 3: brute
*/
signed main()
{
ios::sync_with_stdio(0);
cin.tie(0), cout.tie(0);
cin >> N >> S >> Q;
#ifdef debug
cerr << "debug mode!\n";
vector<int> v(N);
iota(all(v), 1);
mt19937 rd(1);
shuffle(all(v), rd);
stringstream cin;
for (int i = 0; i < N; i++)
cin << v[i] << ' ';
for (int i = 1; i <= Q; i++)
cin << rd() % N + 1 << ' ' << rd() % 1'000'000'000 + 1 << ' ';
cerr << "debug mode end!\n";
stringstream cout;
#endif
for (int i = 1; i <= N; i++)
cin >> p[i], pos[p[i]] = i;
// every query is in the form (n, pos, k, i)
vector<array<int, 4>> qu;
for (int i = 1; i <= Q; i++)
{
int len, k;
cin >> k >> len;
qu.emplace_back(array{N, k, len, i});
}
for (; S >= 1; S--)
{
vector<array<int, 4>> nqu;
sort(all(qu), [&](array<int, 4> a, array<int, 4> b) {return a[0] < b[0]; });
{
int nxt = 1;
blen.init();
for (auto &[n, k, len, i] : qu)
{
while (nxt <= n)
blen.modify(pos[nxt], C(nxt + S - 2, S - 1)), nxt++;
idx[i] = (int)blen.bsearch(len);
len -= blen.query(idx[i]);
}
}
if (S == 1)
{
for (auto &[n, k, pre, i] : qu)
ans[i] += (pos[k] <= idx[i] + 1);
}
else if (S == 2)
{
int nxt = 1;
bcnt.init();
for (auto &[n, k, pre, i] : qu)
{
while (nxt <= n)
bcnt.modify(pos[nxt++], 1);
ans[i] += bcnt.query(idx[i]);
}
sort(all(qu), [&](array<int, 4> a, array<int, 4> b) {return a[1] < b[1]; });
nxt = 1;
bcnt.init();
for (auto &[n, k, pre, i] : qu)
{
while (nxt < k)
bcnt.modify(pos[nxt++], 1);
ans[i] -= bcnt.query(idx[i]);
}
}
else if (S == 3)
{
int nxt = 1;
bcnt.init(), blen.init();
for (auto &[n, k, pre, i] : qu)
{
while (nxt <= n)
bcnt.modify(pos[nxt], 1), blen.modify(pos[nxt], nxt), nxt++;
ans[i] += blen.query(idx[i]) - bcnt.query(idx[i]) * (k - 1);
}
sort(all(qu), [&](array<int, 4> a, array<int, 4> b) {return a[1] < b[1]; });
nxt = 1;
bcnt.init(), blen.init();
for (auto &[n, k, pre, i] : qu)
{
while (nxt < k)
bcnt.modify(pos[nxt], 1), blen.modify(pos[nxt], nxt), nxt++;
ans[i] -= blen.query(idx[i]) - bcnt.query(idx[i]) * (k - 1);
}
}
else
{
int nxt = 1;
set<int> in;
for (auto &[n, k, pre, i] : qu)
{
while (nxt <= n)
in.insert(pos[nxt++]);
assert(idx[i] <= 200);
for (auto j : in)
if (j > idx[i]) break;
else
ans[i] += C(p[j] - k + S - 2, S - 2);
}
}
for (auto &[n, k, pre, i] : qu)
{
if (idx[i] == N || p[idx[i] + 1] < k)
continue;
else
{
tie(n, k, pre) = make_tuple(p[idx[i] + 1], k, pre);
nqu.emplace_back(array{n, k, pre, i});
}
}
nqu.swap(qu);
}
for (int i = 1; i <= Q; i++)
cout << ans[i] << '\n';
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 5616kb
input:
4 3 6 1 4 3 2 1 6 2 20 4 1 3 5 2 9 1 16
output:
3 6 0 1 2 8
result:
ok 6 numbers
Test #2:
score: 0
Accepted
time: 167ms
memory: 19104kb
input:
100000 5 200000 98129 52962 60307 83804 75634 55811 85666 40965 78529 40850 54600 83058 37001 92436 30323 54632 24238 77437 87162 75991 39863 74990 32168 99841 85430 66056 69893 7561 60704 40795 81535 89902 44331 20995 99461 93620 91817 97370 84025 98836 2455 77447 37078 90192 26249 84337 70311 4006...
output:
20217 9885 20204 20217 20218 20217 20217 727 20202 20218 20218 20214 20217 7790 20217 20217 20200 20217 20218 20217 20218 20218 20218 20217 20202 20217 20218 20218 5609 8419 20218 20200 20218 20216 11843 20218 20217 20207 20217 935 20203 5909 20218 20217 20218 20200 20217 20213 20203 20207 5654 4087...
result:
ok 200000 numbers
Test #3:
score: 0
Accepted
time: 148ms
memory: 21072kb
input:
100000 5 200000 59767 75967 97372 50276 20532 77164 30969 40418 84175 87555 79967 28404 26422 22441 4543 72719 9318 93986 12744 88814 25854 93658 9042 41389 24133 60155 80340 44920 58271 50431 92622 28573 30633 318 43850 78809 69750 67699 17767 8454 2543 26572 52552 77502 24901 94119 87156 19368 394...
output:
33305 33330 19455 33332 25500 1777 33332 33310 33337 33333 33297 33331 33337 33330 33317 33304 33332 33336 33331 33337 33335 5611 33334 33337 33333 33331 33337 33307 33334 22446 31995 27049 33337 5351 32269 33332 33334 33331 33299 33331 33334 24072 21451 33331 33338 33332 33331 33331 33319 23356 333...
result:
ok 200000 numbers
Test #4:
score: 0
Accepted
time: 160ms
memory: 19252kb
input:
97997 5 200000 92883 53079 12146 7142 47500 47118 60176 54625 8908 93576 33824 61522 79201 68956 25790 76970 63243 10575 33345 20943 77076 40068 30980 20739 90036 57454 38446 49088 90613 27293 39453 52577 94545 7934 73793 6201 33713 91255 45678 63783 65019 35224 65407 39863 38120 79943 69106 76540 6...
output:
21513 21504 21512 21513 21512 19613 21418 21511 21512 6191 21512 18462 4097 21511 21511 21510 21499 21502 21513 21512 21499 21512 21512 21513 21512 21512 21501 21512 10071 21513 21510 21511 21513 21512 21511 21498 21512 21511 21497 21513 21512 21512 21513 21512 21491 21513 16589 21508 21511 21500 20...
result:
ok 200000 numbers
Test #5:
score: 0
Accepted
time: 175ms
memory: 20608kb
input:
97997 5 200000 1 2 3 4 5 6 7 8 9 10 97997 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 ...
output:
36269 44641 44708 44743 44700 45423 39594 44743 44405 44918 43915 44740 44706 28640 44655 44062 45423 22483 18051 44678 44757 44686 44695 44645 44830 44723 44700 35460 37180 43835 18485 40508 33475 45038 44723 44665 41482 44778 44633 44714 41636 45203 43011 42876 40377 41790 45709 43725 44184 38340 ...
result:
ok 200000 numbers
Test #6:
score: 0
Accepted
time: 131ms
memory: 20568kb
input:
100000 4 200000 57295 10377 65950 19742 78625 70832 98216 11307 9435 8964 76095 1478 46489 61182 52528 10230 50341 83069 94103 42263 18143 52203 16940 68520 76816 98586 96583 28873 59764 20859 85162 22208 33614 52858 48657 87020 40526 40052 31847 30496 89165 52856 35045 71923 20595 57732 33398 21247...
output:
34970 34961 34984 34987 34983 34964 34976 34929 34980 34960 34952 34931 34985 34974 34942 34917 34983 34971 34989 34956 34976 34961 34934 34985 34942 34965 34989 34986 34975 34985 34932 34980 34987 34987 34981 34984 34987 34985 34963 34989 34953 34958 34975 34964 34927 34980 34969 34985 34944 34966 ...
result:
ok 200000 numbers
Test #7:
score: 0
Accepted
time: 162ms
memory: 19748kb
input:
100000 5 200000 65145 22925 57683 48950 47 95014 89971 86419 68041 34514 54368 9070 62968 12058 2395 12953 52181 44707 15141 42970 42586 24500 7627 11589 2549 58411 97779 75300 26698 28015 65395 73656 93413 6314 67115 29279 47001 60377 25606 34502 91013 49546 42758 59346 59698 46487 44285 25907 1322...
output:
10900 0 22894 30657 9289 30701 30702 30701 30695 30695 30392 439 30491 30304 30696 30481 30355 30701 4108 0 30648 24531 0 6293 4639 8441 8856 30702 23474 30417 30634 8671 30702 30703 30703 30287 18522 8635 21217 0 30690 1205 30700 30684 13422 30680 5336 30459 6987 18437 3330 30686 20754 30702 30291 ...
result:
ok 200000 numbers
Test #8:
score: 0
Accepted
time: 75ms
memory: 18304kb
input:
100000 2 200000 58563 67048 17358 71024 31126 36383 27513 94884 69133 65286 93712 27672 17645 50949 98355 2372 66546 2103 98500 62555 45884 64147 34587 58277 2701 14330 80577 48408 94920 43732 23763 39719 18461 34100 35853 73175 64450 77407 54961 17327 5905 22460 81493 59652 22032 32035 61479 57623 ...
output:
7939 20142 20133 20156 20157 20158 20155 6571 17959 20148 20134 20155 18873 20152 20158 20131 20140 20140 20150 20138 20156 20157 20142 20156 20156 20158 20158 20147 20146 20156 20157 20156 20157 20133 20135 20153 20133 20142 20145 20156 20134 20150 20157 12025 20158 20157 20156 20145 20148 20157 18...
result:
ok 200000 numbers
Test #9:
score: 0
Accepted
time: 139ms
memory: 17696kb
input:
1000 5 200000 26 949 329 984 769 794 982 993 558 219 670 13 247 251 931 678 557 142 24 539 903 787 436 731 536 29 579 963 365 87 546 912 592 620 129 637 676 945 588 386 438 482 300 43 67 205 274 68 551 71 478 95 260 458 636 414 368 953 586 856 944 914 114 853 444 621 263 631 605 187 201 960 94 764 5...
output:
3703241 4008426 4089839 4001808 3246557 4014241 3989395 391064 4059813 4001808 3784896 4021324 3700469 4077139 4089854 3820273 3588482 3151986 4077116 4014232 3915999 3079302 3820273 4073494 4089846 3791425 4001808 3691479 4051829 3977074 4013302 2581861 3268414 1855291 3726455 4084881 4064457 40267...
result:
ok 200000 numbers
Test #10:
score: 0
Accepted
time: 29ms
memory: 17972kb
input:
1000 1 200000 301 253 940 560 437 708 946 42 29 56 398 103 526 147 238 264 613 507 635 734 110 625 220 186 795 790 189 306 755 923 980 579 572 22 577 530 833 574 722 803 954 159 644 700 657 761 806 225 300 179 471 951 463 1 217 889 379 43 739 709 976 139 817 32 837 243 486 152 950 182 675 972 666 36...
output:
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 0 1 1 1 1 1 1 1 1 1 1 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 ...
result:
ok 200000 numbers
Test #11:
score: 0
Accepted
time: 1ms
memory: 7588kb
input:
3 3 3 2 3 1 1 1 2 5 1 2
output:
0 2 1
result:
ok 3 number(s): "0 2 1"