結果

問題 No.1856 Mex Sum 2
ユーザー qwewe
提出日時 2025-04-24 12:27:34
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 750 bytes
コンパイル時間 213 ms
コンパイル使用メモリ 81,712 KB
実行使用メモリ 91,176 KB
最終ジャッジ日時 2025-04-24 12:29:38
合計ジャッジ時間 26,436 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 45 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

N, M = map(int, input().split())

# Precompute binomial coefficients C[k][t] for 0 <= t <=k <=N
C = []
for k in range(N + 1):
    row = [0] * (k + 1)
    row[0] = 1
    if k >= 1:
        row[k] = 1
    for t in range(1, k):
        row[t] = (C[k-1][t-1] + C[k-1][t]) % MOD
    C.append(row)

M_mod = M % MOD

ans = 0
for k in range(1, N + 1):
    sum_k = 0
    base = (2 * M_mod + 1) % MOD
    for t in range(0, k + 1):
        current_base = (base - t) % MOD
        pow_val = pow(current_base, N, MOD)
        sign = 1 if t % 2 == 0 else MOD - 1
        term = (sign * C[k][t]) % MOD
        term = (term * pow_val) % MOD
        sum_k = (sum_k + term) % MOD
    sum_k = (sum_k * k) % MOD
    ans = (ans + sum_k) % MOD

print(ans)
0