結果
問題 | No.2636 No Waiting in Vain |
ユーザー |
![]() |
提出日時 | 2024-02-18 13:09:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 92 ms / 2,000 ms |
コード長 | 731 bytes |
コンパイル時間 | 389 ms |
コンパイル使用メモリ | 82,592 KB |
実行使用メモリ | 98,304 KB |
最終ジャッジ日時 | 2024-09-29 00:23:00 |
合計ジャッジ時間 | 4,591 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 44 |
ソースコード
N, K = map(int, input().split())S = list(input())mod = 998244353n = 505050fact = [1] * (n + 1)invfact = [1] * (n + 1)for i in range(1, n):fact[i + 1] = ((i+1) * fact[i]) % modinvfact[n] = pow(fact[n], mod - 2, mod)for i in range(n - 1, -1, -1):invfact[i] = invfact[i + 1] * (i + 1) % moddef comb(n, r):if n < 0 or r < 0 or n - r < 0:return 0return fact[n] * invfact[r] * invfact[n - r] % modcnt = 0flag = 0for s in S:if s == "C":flag = 1elif s == "N":if not flag:cnt += 1else:K -= 1flag = 0else:flag = 0ans = 0for i in range(1, cnt - K + 1):ans += comb(cnt, i)ans %= modprint(ans)