結果
問題 |
No.2636 No Waiting in Vain
|
ユーザー |
![]() |
提出日時 | 2024-02-23 10:57:00 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 70 ms / 2,000 ms |
コード長 | 637 bytes |
コンパイル時間 | 318 ms |
コンパイル使用メモリ | 82,428 KB |
実行使用メモリ | 76,492 KB |
最終ジャッジ日時 | 2024-09-29 05:00:25 |
合計ジャッジ時間 | 3,950 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 44 |
ソースコード
N, K = map(int, input().split()) MOD = 998244353 fact = [1] * (N + 1) rfact = [1] * (N + 1) r = 1 for i in range(1, N + 1): fact[i] = r = r * i % MOD rfact[N] = r = pow(fact[N], MOD - 2, MOD) for i in range(N, 0, -1): rfact[i - 1] = r = r * i % MOD # nCk (mod MOD) を求める def comb(n, k): return fact[n] * rfact[k] * rfact[n - k] % MOD S = input() n = 0 cn = 0 for i in range(N): if S[i] == "N": if i > 0 and S[i - 1] == "C": cn += 1 else: n += 1 K = max(K - cn, 0) ans = 0 if n > K: for i in range(1, n - K + 1): ans += comb(n, i) ans %= MOD print(ans)