結果

問題 No.2636 No Waiting in Vain
ユーザー 👑 loop0919
提出日時 2024-02-18 13:36:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 397 ms / 2,000 ms
コード長 534 bytes
コンパイル時間 393 ms
コンパイル使用メモリ 82,204 KB
実行使用メモリ 133,880 KB
最終ジャッジ日時 2024-09-29 00:24:08
合計ジャッジ時間 10,268 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353

N, K = map(int, input().split())
S = ['P'] + list(input())

# 階乗全列挙
facts = [1]
inv_facts = [1]

for i in range(1, N+1):
    facts.append(facts[i-1] * i % MOD)
    inv_facts.append(inv_facts[i-1] * pow(i, MOD-2, MOD) % MOD)

# nCr 全列挙
def comb(n, r):
    return facts[n] * inv_facts[n-r] * inv_facts[r] % MOD

max_plan = S.count('N') - K

cnt = 0

for i in range(1, N+1):
    if S[i] == 'N' and S[i-1] != 'C':
        cnt += 1

print(sum(comb(cnt, i) for i in range(1, min(max_plan, cnt) + 1)) % MOD)
0