QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#27853#2618. Casual DancersGeothermal#AC ✓121ms25972kbC++5.0kb2022-04-11 02:28:022022-04-29 07:49:37

Judging History

This is the latest submission verdict.

  • [2023-08-10 23:21:45]
  • System Update: QOJ starts to keep a history of the judgings of all the submissions.
  • [2022-04-29 07:49:37]
  • Judged
  • Verdict: AC
  • Time: 121ms
  • Memory: 25972kb
  • [2022-04-11 02:28:02]
  • Submitted

answer

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const ll modi = 998244353;

ll modExp(ll base,ll times){
    ll ans = 1;
    while (times){
        if (times & 1) ans = ans * base % modi;
        base = base * base % modi;
        times >>= 1;
    }
    return ans;
}

ll inv(ll x){
    return modExp(x,modi-2);
}

namespace NTT{
    vector<ll> roots = {0,1};
    vector<int> bit_reverse;
    int max_size = -1;
    ll root;
    bool is_power_of_two(int n){
        return (n & (n-1)) == 0;
    }
    int round_up_power_two(int n){
        while (n & (n-1)){
            n = (n | (n-1))+1;
        }
        return max(n,1);
    }
    int get_length(int n){
        assert(is_power_of_two(n));
        return __builtin_ctz(n);
    }
    void bit_reorder(int n,vector<ll> & values){
        if ((int) bit_reverse.size() != n){
            bit_reverse.assign(n,0);
            int length = get_length(n);
            for (int i = 0;i < n;i++){
                bit_reverse[i] = (bit_reverse[i >> 1] >> 1) + ((i & 1) << (length - 1));
            }
        }
        for (int i = 0;i < n;i++){
            if (i < bit_reverse[i]){
                swap(values[i], values[bit_reverse[i]]);
            }
        }
    }
    void find_root(){
        max_size = 1 << __builtin_ctz(modi-1);
        root = 2;
        while (!(modExp(root,max_size) == 1 && modExp(root, max_size/2) != 1)) root++;
    }
    void prepare_roots(int n){
        if (max_size < 0) find_root();
        assert (n <= max_size);
        if ((int)roots.size() >= n) return;
        int length = get_length(roots.size());
        roots.resize(n);
        while (1 << length < n){
            ll z = modExp(root,max_size >> (length+1));
            for (int i = 1 << (length - 1); i < 1 << length;i++){
                roots[2*i] = roots[i];
                roots[2*i+1] = (roots[i] * z) % modi;
            }
            length++;
        }
    }
    void fft_iterative(int N,vector<ll> & values){
        assert(is_power_of_two(N));
        prepare_roots(N);
        bit_reorder(N, values);
        for (int n = 1;n < N;n *= 2){
            for (int start = 0; start < N;start += 2*n){
                for (int i = 0;i < n;i++){
                    ll even = values[start+i];
                    ll odd = values[start+n+i] * roots[n+i];
                    odd %= modi;
                    values[start+n+i] = even-odd+modi;
                    values[start+i] = even+odd;
                    values[start+n+i] %= modi;
                    values[start+i] %= modi;
                }
            }
        }
    }
    const int FFT_CUTOFF = 150;
    vector<ll> mod_multiply(vector<ll> left,vector<ll> right){
        int n = left.size();
        int m = right.size();
        if (min(n,m) < FFT_CUTOFF){
            const uint64_t ULL_BOUND = numeric_limits<uint64_t>::max() - (uint64_t) modi*modi;
            vector<uint64_t> result(n+m-1,0);
            for (int i = 0;i < n;i++){
                for (int j = 0;j < m;j++){
                    result[i+j] += (uint64_t)((int)left[i])*((int)right[j]);
                    if (result[i+j] > ULL_BOUND) result[i+j] %= modi;
                }
            }
            for (uint64_t & x: result){
                if (x >= modi) x %= modi;
            }
            return vector<ll>(result.begin(),result.end());
        }
        int N = round_up_power_two(n+m-1);
        left.resize(N);
        right.resize(N);
        bool equal = left == right;
        fft_iterative(N,left);
        if (equal) right = left;
        else fft_iterative(N,right);
        ll inv_N = inv(N);
        for (int i = 0;i < N;i++){
            left[i] *= (right[i] * inv_N) % modi;
            left[i] %= modi;
        }
        reverse(left.begin()+1,left.end());
        fft_iterative(N,left);
        left.resize(n+m-1);
        return left;
    }
    vector<ll> mod_power(const vector<ll> & v,int exponent){
        assert(exponent>=0);
        vector<ll> result = {1};
        if (exponent == 0) return result;
        for (int k = 31 - __builtin_clz(exponent);k >= 0;k--){
            result = mod_multiply(result,result);
            if (exponent >> k & 1){
                result = mod_multiply(result,v);
            }
        }
        return result;
    }
}


int main(){
    int x1,x2,x3;
    int k;
    ll p;
    cin >> x1 >> x2 >> x3 >> k >> p;
    vector<ll> poly = {1,1,1};
    vector<ll> ret = NTT::mod_power(poly, k);
    ll e = 0;

    ll delta = x1-x2;
    for (int i = 0;i <= 2*k;i++){
        ll new_delta = delta + i - k;
        e = (e + abs(new_delta) * ret[i]) % modi;
    }
    delta = x1-x3;
    for (int i = 0;i <= 2*k;i++){
        ll new_delta = delta + i - k;
        e = (e + abs(new_delta) * ret[i]) % modi;
    }
    delta = x2-x3;
    for (int i = 0;i <= 2*k;i++){
        ll new_delta = delta + i - k;
        e = (e + abs(new_delta) * ret[i]) % modi;
    }
    cout << e * inv(modExp(3,k)) % modi * inv(2) % modi << endl;
}

詳細信息

Test #1:

score: 100
Accepted
time: 1ms
memory: 3640kb

input:

0 0 0
1
58

output:

1

result:

ok 1 number(s): "1"

Test #2:

score: 0
Accepted
time: 0ms
memory: 3704kb

input:

1 2 2
1
100

output:

332748119

result:

ok 1 number(s): "332748119"

Test #3:

score: 0
Accepted
time: 3ms
memory: 3612kb

input:

5 2 3
4
50

output:

160212060

result:

ok 1 number(s): "160212060"

Test #4:

score: 0
Accepted
time: 3ms
memory: 3508kb

input:

-2 -1 1
2
71

output:

443664160

result:

ok 1 number(s): "443664160"

Test #5:

score: 0
Accepted
time: 0ms
memory: 3652kb

input:

-1 0 -1
4
8

output:

751764268

result:

ok 1 number(s): "751764268"

Test #6:

score: 0
Accepted
time: 2ms
memory: 3564kb

input:

-2 -2 2
5
54

output:

801060288

result:

ok 1 number(s): "801060288"

Test #7:

score: 0
Accepted
time: 3ms
memory: 3712kb

input:

-2 2 4
8
36

output:

353135983

result:

ok 1 number(s): "353135983"

Test #8:

score: 0
Accepted
time: 3ms
memory: 3504kb

input:

8 -7 1
7
28

output:

15

result:

ok 1 number(s): "15"

Test #9:

score: 0
Accepted
time: 3ms
memory: 3632kb

input:

-7 -8 0
16
54

output:

159363041

result:

ok 1 number(s): "159363041"

Test #10:

score: 0
Accepted
time: 2ms
memory: 3608kb

input:

5 11 -11
9
32

output:

717226646

result:

ok 1 number(s): "717226646"

Test #11:

score: 0
Accepted
time: 1ms
memory: 3608kb

input:

-16 1 -9
32
8

output:

855967855

result:

ok 1 number(s): "855967855"

Test #12:

score: 0
Accepted
time: 3ms
memory: 3612kb

input:

-13 28 28
37
80

output:

116405794

result:

ok 1 number(s): "116405794"

Test #13:

score: 0
Accepted
time: 3ms
memory: 3580kb

input:

6 26 -25
64
21

output:

91053409

result:

ok 1 number(s): "91053409"

Test #14:

score: 0
Accepted
time: 3ms
memory: 3560kb

input:

-39 23 1
31
64

output:

742331784

result:

ok 1 number(s): "742331784"

Test #15:

score: 0
Accepted
time: 3ms
memory: 3716kb

input:

-32 42 43
128
87

output:

57822539

result:

ok 1 number(s): "57822539"

Test #16:

score: 0
Accepted
time: 3ms
memory: 3656kb

input:

-80 55 -106
142
29

output:

435655440

result:

ok 1 number(s): "435655440"

Test #17:

score: 0
Accepted
time: 3ms
memory: 3680kb

input:

0 -83 -106
256
55

output:

120508896

result:

ok 1 number(s): "120508896"

Test #18:

score: 0
Accepted
time: 3ms
memory: 3708kb

input:

-100 -123 -167
91
74

output:

285780715

result:

ok 1 number(s): "285780715"

Test #19:

score: 0
Accepted
time: 3ms
memory: 3796kb

input:

252 -176 -239
512
49

output:

835642397

result:

ok 1 number(s): "835642397"

Test #20:

score: 0
Accepted
time: 3ms
memory: 3724kb

input:

-37 -124 151
867
76

output:

225290884

result:

ok 1 number(s): "225290884"

Test #21:

score: 0
Accepted
time: 3ms
memory: 3812kb

input:

-316 149 -149
1024
87

output:

374987754

result:

ok 1 number(s): "374987754"

Test #22:

score: 0
Accepted
time: 0ms
memory: 3760kb

input:

370 545 81
1073
69

output:

943329809

result:

ok 1 number(s): "943329809"

Test #23:

score: 0
Accepted
time: 0ms
memory: 3984kb

input:

-81 182 532
2048
87

output:

843173062

result:

ok 1 number(s): "843173062"

Test #24:

score: 0
Accepted
time: 3ms
memory: 3676kb

input:

-1229 -1607 319
199
24

output:

1926

result:

ok 1 number(s): "1926"

Test #25:

score: 0
Accepted
time: 2ms
memory: 3972kb

input:

43 -419 -613
4096
46

output:

418220629

result:

ok 1 number(s): "418220629"

Test #26:

score: 0
Accepted
time: 1ms
memory: 3812kb

input:

3434 -3146 -1774
2601
46

output:

705802517

result:

ok 1 number(s): "705802517"

Test #27:

score: 0
Accepted
time: 5ms
memory: 4644kb

input:

2193 -2331 2901
8192
75

output:

728593792

result:

ok 1 number(s): "728593792"

Test #28:

score: 0
Accepted
time: 3ms
memory: 3840kb

input:

233 -4307 -4363
1093
81

output:

303899847

result:

ok 1 number(s): "303899847"

Test #29:

score: 0
Accepted
time: 12ms
memory: 5568kb

input:

-4522 762 8059
16384
34

output:

190696426

result:

ok 1 number(s): "190696426"

Test #30:

score: 0
Accepted
time: 11ms
memory: 5904kb

input:

-5155 -3639 15798
24822
55

output:

808461103

result:

ok 1 number(s): "808461103"

Test #31:

score: 0
Accepted
time: 26ms
memory: 8300kb

input:

15234 4368 12248
32768
19

output:

115861480

result:

ok 1 number(s): "115861480"

Test #32:

score: 0
Accepted
time: 16ms
memory: 8404kb

input:

820 30492 3951
42789
76

output:

826647308

result:

ok 1 number(s): "826647308"

Test #33:

score: 0
Accepted
time: 48ms
memory: 13492kb

input:

1372 -24835 -24597
65536
65

output:

355997764

result:

ok 1 number(s): "355997764"

Test #34:

score: 0
Accepted
time: 56ms
memory: 13876kb

input:

-59726 17559 -45875
87143
58

output:

326130350

result:

ok 1 number(s): "326130350"

Test #35:

score: 0
Accepted
time: 109ms
memory: 23712kb

input:

-27584 51950 23030
131072
74

output:

325794325

result:

ok 1 number(s): "325794325"

Test #36:

score: 0
Accepted
time: 107ms
memory: 24092kb

input:

61155 52006 74974
164861
5

output:

160748350

result:

ok 1 number(s): "160748350"

Test #37:

score: 0
Accepted
time: 102ms
memory: 24592kb

input:

41344 -81596 -95774
200000
59

output:

965482998

result:

ok 1 number(s): "965482998"

Test #38:

score: 0
Accepted
time: 16ms
memory: 5336kb

input:

42056 -90767 -54649
24350
63

output:

132823

result:

ok 1 number(s): "132823"

Test #39:

score: 0
Accepted
time: 111ms
memory: 25944kb

input:

-74335 43393 57021
199994
67

output:

310210583

result:

ok 1 number(s): "310210583"

Test #40:

score: 0
Accepted
time: 112ms
memory: 23492kb

input:

-80838 73772 -18618
134415
57

output:

346157175

result:

ok 1 number(s): "346157175"

Test #41:

score: 0
Accepted
time: 112ms
memory: 24672kb

input:

37457 74497 -81166
199997
59

output:

26667908

result:

ok 1 number(s): "26667908"

Test #42:

score: 0
Accepted
time: 110ms
memory: 24164kb

input:

31109 -65140 -77085
162412
46

output:

12858959

result:

ok 1 number(s): "12858959"

Test #43:

score: 0
Accepted
time: 121ms
memory: 25972kb

input:

-58550 -97769 66373
199995
86

output:

789346262

result:

ok 1 number(s): "789346262"

Test #44:

score: 0
Accepted
time: 59ms
memory: 15336kb

input:

7739 58831 72332
124270
16

output:

167162440

result:

ok 1 number(s): "167162440"

Test #45:

score: 0
Accepted
time: 120ms
memory: 25888kb

input:

-97901 25173 -99145
199999
52

output:

797290311

result:

ok 1 number(s): "797290311"

Test #46:

score: 0
Accepted
time: 53ms
memory: 15512kb

input:

-87118 -60882 -86669
126103
23

output:

487838027

result:

ok 1 number(s): "487838027"

Test #47:

score: 0
Accepted
time: 97ms
memory: 24592kb

input:

-71646 69885 70206
200000
27

output:

285981891

result:

ok 1 number(s): "285981891"

Test #48:

score: 0
Accepted
time: 46ms
memory: 14632kb

input:

14475 -77173 -5177
117777
51

output:

251933976

result:

ok 1 number(s): "251933976"

Test #49:

score: 0
Accepted
time: 107ms
memory: 24600kb

input:

-35780 37165 54712
199996
14

output:

763964192

result:

ok 1 number(s): "763964192"

Test #50:

score: 0
Accepted
time: 57ms
memory: 13996kb

input:

15709 -72676 -22298
101968
17

output:

406652317

result:

ok 1 number(s): "406652317"

Test #51:

score: 0
Accepted
time: 111ms
memory: 25876kb

input:

74572 -98701 -56974
199991
62

output:

55467556

result:

ok 1 number(s): "55467556"

Test #52:

score: 0
Accepted
time: 112ms
memory: 24948kb

input:

-14644 -10031 -50353
168383
43

output:

376814948

result:

ok 1 number(s): "376814948"

Test #53:

score: 0
Accepted
time: 112ms
memory: 25860kb

input:

22388 51898 80903
199995
89

output:

832434478

result:

ok 1 number(s): "832434478"

Test #54:

score: 0
Accepted
time: 52ms
memory: 15136kb

input:

34062 -76211 -25545
127193
91

output:

234760702

result:

ok 1 number(s): "234760702"

Test #55:

score: 0
Accepted
time: 106ms
memory: 24724kb

input:

-19645 -45450 -16512
200000
77

output:

759439547

result:

ok 1 number(s): "759439547"

Test #56:

score: 0
Accepted
time: 14ms
memory: 5820kb

input:

98957 80512 -24606
20311
30

output:

985804570

result:

ok 1 number(s): "985804570"

Test #57:

score: 0
Accepted
time: 121ms
memory: 25920kb

input:

-87259 -11505 14596
199994
83

output:

160520754

result:

ok 1 number(s): "160520754"

Test #58:

score: 0
Accepted
time: 110ms
memory: 24676kb

input:

0 0 0
200000
0

output:

393458944

result:

ok 1 number(s): "393458944"

Test #59:

score: 0
Accepted
time: 116ms
memory: 24724kb

input:

0 0 0
200000
100

output:

393458944

result:

ok 1 number(s): "393458944"

Test #60:

score: 0
Accepted
time: 109ms
memory: 24548kb

input:

-100000 -100000 -100000
200000
75

output:

393458944

result:

ok 1 number(s): "393458944"

Test #61:

score: 0
Accepted
time: 106ms
memory: 24620kb

input:

100000 100000 100000
200000
63

output:

393458944

result:

ok 1 number(s): "393458944"

Test #62:

score: 0
Accepted
time: 97ms
memory: 24712kb

input:

100000 0 -100000
200000
56

output:

678255914

result:

ok 1 number(s): "678255914"

Test #63:

score: 0
Accepted
time: 109ms
memory: 24644kb

input:

0 1 2
200000
22

output:

630634769

result:

ok 1 number(s): "630634769"

Test #64:

score: 0
Accepted
time: 3ms
memory: 3632kb

input:

100000 0 -100000
1
32

output:

200000

result:

ok 1 number(s): "200000"

Test #65:

score: 0
Accepted
time: 3ms
memory: 3636kb

input:

100000 100000 100000
1
33

output:

1

result:

ok 1 number(s): "1"

Test #66:

score: 0
Accepted
time: 1ms
memory: 3700kb

input:

-100000 -100000 -100000
1
6

output:

1

result:

ok 1 number(s): "1"

Test #67:

score: 0
Accepted
time: 3ms
memory: 3560kb

input:

100000 100000 -100000
1
7

output:

332948118

result:

ok 1 number(s): "332948118"

Test #68:

score: 0
Accepted
time: 3ms
memory: 3560kb

input:

-100000 -100000 100000
1
40

output:

332948118

result:

ok 1 number(s): "332948118"

Test #69:

score: 0
Accepted
time: 3ms
memory: 3656kb

input:

100000 -100000 -100000
100
63

output:

764105630

result:

ok 1 number(s): "764105630"

Test #70:

score: 0
Accepted
time: 1ms
memory: 3568kb

input:

-100000 100000 100000
100
13

output:

764105630

result:

ok 1 number(s): "764105630"

Test #71:

score: 0
Accepted
time: 3ms
memory: 3572kb

input:

-100000 100000 0
100
10

output:

200000

result:

ok 1 number(s): "200000"

Test #72:

score: 0
Accepted
time: 66ms
memory: 14508kb

input:

-100000 100000 0
99999
77

output:

200000

result:

ok 1 number(s): "200000"

Test #73:

score: 0
Accepted
time: 46ms
memory: 13920kb

input:

-100000 100000 0
100000
80

output:

200000

result:

ok 1 number(s): "200000"

Test #74:

score: 0
Accepted
time: 49ms
memory: 13844kb

input:

-100000 100000 0
100001
5

output:

50125708

result:

ok 1 number(s): "50125708"