QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#525414#1823. Permutation CFGRngBased#AC ✓175ms21072kbC++174.9kb2024-08-20 16:16:102024-08-20 16:16:12

Judging History

你现在查看的是最新测评结果

  • [2024-08-20 16:16:12]
  • 评测
  • 测评结果:AC
  • 用时:175ms
  • 内存:21072kb
  • [2024-08-20 16:16:10]
  • 提交

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"