QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#350455#8171. Colaucup-team2307#AC ✓590ms393908kbC++142.3kb2024-03-10 18:47:312024-03-10 18:47:31

Judging History

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

  • [2024-03-10 18:47:31]
  • 评测
  • 测评结果:AC
  • 用时:590ms
  • 内存:393908kb
  • [2024-03-10 18:47:31]
  • 提交

answer

#pragma GCC optimize("trapv")
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int mod = 998244353;
int inv(int x) {
    return x < 2 ? x : mod - (mod / x) * 1ll * inv(mod % x) % mod;
}
vector<ll> get_pent(int n) {
    vector<ll> penta(n + 1);
    penta[0] = 1;
    penta[1] = -1;
    for(ll x = 0; x <= n; x++) {
        ll a = (3*x*x+7*x+4) / 2;
        ll b = a + 2*x + 3;
        // cout << a << " " << b << endl;
        if(a > n && b > n) break;
        if(a <= n) penta[a] = x % 2 ? 1 : -1;
        if(b <= n) penta[b] = (x+1) % 2 ? 1 : -1;
    }
    return penta;
}
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n, m;
    cin >> n >> m;
    // vector<ll> dp(m); dp[0] = 1;
    // for(int k = n; k >= 1; k--) {
    //     int X = inv(k) % mod;
    //     vector<ll> ndp(m);
    //     for(int i = m; i--;) {
    //         for(int t = 0; t < k && i - t >= 0; t++)
    //             ndp[i] = (ndp[i] + dp[i - t] * X) % mod;
    //     }
    //     dp = ndp;
    // }
    // ll fact = 1, v = 0;
    // for(int i = 0; i < m; i++) v = (v + dp[i]) % mod;
    // cout << v << endl;
    // vector<ll> dp(n + 1);dp[0] = 1;
    // for(int i = 1; i <= n; i++) {
    //     for(int j = n; j - i >= 0; j--)
    //         dp[j] = (dp[j] + mod - dp[j - i]) % mod;
    // }
    auto dp = get_pent(n + 1);

    vector<ll> fact(2*n + 1, 1), ifact(2*n + 1, 1);
    for(int i = 2; i <= 2*n; i++) {
        ifact[i] = mod - (mod / i) * ifact[mod % i] % mod;
    }
    for(int i = 2; i <= 2*n; i++) {
        fact[i] = fact[i-1] * 1ll * i % mod;
        ifact[i] = ifact[i - 1] * 1ll * ifact[i] % mod;
    }
    auto C = [&](int n, int k) -> ll{
        if(k < 0 || k > n) return 0;
        ll res = fact[n] * ifact[k] % mod;
        return res * ifact[n - k] % mod;
    };

    int T = n + 1;
    // for(int T = n + 1; T--;) {
    //     for(int i = 1; i <= n; i++) {
    //         // assert(i < dp.size());
    //         (dp[i] += dp[i - 1]) %=mod;
    //     }
    // }
    ll v = 0;m--;
    for(int i = 0; i <= m; i++) {
        // cout << C(i+T-1, i) << " " << dp[m - i] << endl;
        v = (v + C(i+T-1, i) * dp[m - i]) % mod;
    }
    v = v * ifact[n] % mod;
    v = (mod + v) % mod;
    cout << v << endl;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3536kb

input:

2 1

output:

499122177

result:

ok "499122177"

Test #2:

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

input:

1 1

output:

1

result:

ok "1"

Test #3:

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

input:

167 91

output:

469117530

result:

ok "469117530"

Test #4:

score: 0
Accepted
time: 532ms
memory: 386120kb

input:

9806463 8975779

output:

125384417

result:

ok "125384417"

Test #5:

score: 0
Accepted
time: 516ms
memory: 360000kb

input:

9138576 8731432

output:

306972756

result:

ok "306972756"

Test #6:

score: 0
Accepted
time: 548ms
memory: 392948kb

input:

9978791 9033584

output:

932159263

result:

ok "932159263"

Test #7:

score: 0
Accepted
time: 561ms
memory: 386512kb

input:

9811954 9790000

output:

404679920

result:

ok "404679920"

Test #8:

score: 0
Accepted
time: 532ms
memory: 381388kb

input:

9685105 9276909

output:

32996715

result:

ok "32996715"

Test #9:

score: 0
Accepted
time: 589ms
memory: 393736kb

input:

10000000 10000000

output:

309225852

result:

ok "309225852"

Test #10:

score: 0
Accepted
time: 590ms
memory: 393836kb

input:

10000000 9999999

output:

635234302

result:

ok "635234302"

Test #11:

score: 0
Accepted
time: 567ms
memory: 393904kb

input:

10000000 9999998

output:

239117935

result:

ok "239117935"

Test #12:

score: 0
Accepted
time: 562ms
memory: 393868kb

input:

10000000 9999997

output:

294859983

result:

ok "294859983"

Test #13:

score: 0
Accepted
time: 564ms
memory: 393784kb

input:

9999999 9999999

output:

305530110

result:

ok "305530110"

Test #14:

score: 0
Accepted
time: 568ms
memory: 393812kb

input:

9999999 9999998

output:

164959553

result:

ok "164959553"

Test #15:

score: 0
Accepted
time: 575ms
memory: 393772kb

input:

9999999 9999997

output:

532215262

result:

ok "532215262"

Test #16:

score: 0
Accepted
time: 563ms
memory: 393768kb

input:

9999999 9999996

output:

123628609

result:

ok "123628609"

Test #17:

score: 0
Accepted
time: 573ms
memory: 393732kb

input:

9999998 9999998

output:

223852357

result:

ok "223852357"

Test #18:

score: 0
Accepted
time: 581ms
memory: 393768kb

input:

9999998 9999997

output:

75877991

result:

ok "75877991"

Test #19:

score: 0
Accepted
time: 564ms
memory: 393904kb

input:

9999998 9999996

output:

494540335

result:

ok "494540335"

Test #20:

score: 0
Accepted
time: 568ms
memory: 393820kb

input:

9999998 9999995

output:

19191738

result:

ok "19191738"

Test #21:

score: 0
Accepted
time: 570ms
memory: 393828kb

input:

9999997 9999997

output:

238385746

result:

ok "238385746"

Test #22:

score: 0
Accepted
time: 552ms
memory: 393900kb

input:

9999997 9999996

output:

138191521

result:

ok "138191521"

Test #23:

score: 0
Accepted
time: 556ms
memory: 393740kb

input:

9999997 9999995

output:

721536184

result:

ok "721536184"

Test #24:

score: 0
Accepted
time: 541ms
memory: 393908kb

input:

9999997 9999994

output:

627112720

result:

ok "627112720"

Test #25:

score: 0
Accepted
time: 373ms
memory: 320124kb

input:

8113616 1826492

output:

629546539

result:

ok "629546539"

Test #26:

score: 0
Accepted
time: 364ms
memory: 285564kb

input:

7230333 4233627

output:

870135249

result:

ok "870135249"

Test #27:

score: 0
Accepted
time: 536ms
memory: 383476kb

input:

9734872 9617286

output:

780426509

result:

ok "780426509"

Test #28:

score: 0
Accepted
time: 373ms
memory: 267840kb

input:

6780022 6393958

output:

508662111

result:

ok "508662111"

Test #29:

score: 0
Accepted
time: 221ms
memory: 198000kb

input:

4986441 1909344

output:

762587564

result:

ok "762587564"

Test #30:

score: 0
Accepted
time: 445ms
memory: 391192kb

input:

9936540 91728

output:

651924678

result:

ok "651924678"

Test #31:

score: 0
Accepted
time: 394ms
memory: 358712kb

input:

9099529 94239

output:

775532638

result:

ok "775532638"

Test #32:

score: 0
Accepted
time: 433ms
memory: 376636kb

input:

9564814 93545

output:

474538902

result:

ok "474538902"

Test #33:

score: 0
Accepted
time: 453ms
memory: 382404kb

input:

9707744 92094

output:

354024226

result:

ok "354024226"

Test #34:

score: 0
Accepted
time: 401ms
memory: 361344kb

input:

9167687 94820

output:

858989558

result:

ok "858989558"

Test #35:

score: 0
Accepted
time: 456ms
memory: 393740kb

input:

10000000 100000

output:

609345536

result:

ok "609345536"

Test #36:

score: 0
Accepted
time: 437ms
memory: 393776kb

input:

10000000 99999

output:

217258255

result:

ok "217258255"

Test #37:

score: 0
Accepted
time: 438ms
memory: 393724kb

input:

10000000 99998

output:

485057696

result:

ok "485057696"

Test #38:

score: 0
Accepted
time: 417ms
memory: 393752kb

input:

10000000 99997

output:

193579142

result:

ok "193579142"

Test #39:

score: 0
Accepted
time: 415ms
memory: 393740kb

input:

9999999 100000

output:

584105896

result:

ok "584105896"

Test #40:

score: 0
Accepted
time: 423ms
memory: 393736kb

input:

9999999 99999

output:

707014865

result:

ok "707014865"

Test #41:

score: 0
Accepted
time: 420ms
memory: 393768kb

input:

9999999 99998

output:

872987417

result:

ok "872987417"

Test #42:

score: 0
Accepted
time: 431ms
memory: 393908kb

input:

9999999 99997

output:

707304988

result:

ok "707304988"

Test #43:

score: 0
Accepted
time: 426ms
memory: 393772kb

input:

9999998 100000

output:

789028925

result:

ok "789028925"

Test #44:

score: 0
Accepted
time: 436ms
memory: 393832kb

input:

9999998 99999

output:

628266237

result:

ok "628266237"

Test #45:

score: 0
Accepted
time: 432ms
memory: 393892kb

input:

9999998 99998

output:

38358057

result:

ok "38358057"

Test #46:

score: 0
Accepted
time: 428ms
memory: 393756kb

input:

9999998 99997

output:

785931240

result:

ok "785931240"

Test #47:

score: 0
Accepted
time: 422ms
memory: 393740kb

input:

9999997 100000

output:

945452715

result:

ok "945452715"

Test #48:

score: 0
Accepted
time: 416ms
memory: 393736kb

input:

9999997 99999

output:

537126025

result:

ok "537126025"

Test #49:

score: 0
Accepted
time: 430ms
memory: 393832kb

input:

9999997 99998

output:

837196653

result:

ok "837196653"

Test #50:

score: 0
Accepted
time: 425ms
memory: 393892kb

input:

9999997 99997

output:

263045713

result:

ok "263045713"

Test #51:

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

input:

2 2

output:

1

result:

ok "1"

Test #52:

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

input:

3 3

output:

831870295

result:

ok "831870295"

Test #53:

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

input:

4 4

output:

374341633

result:

ok "374341633"

Test #54:

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

input:

3 2

output:

499122177

result:

ok "499122177"

Test #55:

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

input:

4 3

output:

623902721

result:

ok "623902721"

Test #56:

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

input:

5 4

output:

890101215

result:

ok "890101215"

Test #57:

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

input:

3 1

output:

166374059

result:

ok "166374059"

Test #58:

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

input:

4 2

output:

166374059

result:

ok "166374059"

Test #59:

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

input:

5 3

output:

16637406

result:

ok "16637406"

Test #60:

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

input:

6 4

output:

508827330

result:

ok "508827330"

Test #61:

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

input:

4 1

output:

291154603

result:

ok "291154603"

Test #62:

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

input:

5 2

output:

291154603

result:

ok "291154603"

Test #63:

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

input:

6 3

output:

859599304

result:

ok "859599304"

Test #64:

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

input:

7 4

output:

694809760

result:

ok "694809760"

Test #65:

score: 0
Accepted
time: 225ms
memory: 222932kb

input:

5629201 38642

output:

327294391

result:

ok "327294391"

Test #66:

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

input:

126092 74219

output:

27573951

result:

ok "27573951"

Test #67:

score: 0
Accepted
time: 343ms
memory: 338960kb

input:

8593075 8689

output:

393785113

result:

ok "393785113"

Test #68:

score: 0
Accepted
time: 36ms
memory: 45392kb

input:

1076972 58637

output:

600806929

result:

ok "600806929"

Test #69:

score: 0
Accepted
time: 13ms
memory: 21284kb

input:

463217 39187

output:

408712022

result:

ok "408712022"

Extra Test:

score: 0
Extra Test Passed