結果

問題 No.2388 At Least K-Characters
ユーザー 👑 Mizar
提出日時 2023-07-13 01:44:28
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 387 ms / 4,000 ms
コード長 1,213 bytes
コンパイル時間 300 ms
コンパイル使用メモリ 81,844 KB
実行使用メモリ 77,356 KB
最終ジャッジ日時 2024-07-05 03:39:13
合計ジャッジ時間 9,732 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

n, m, k = map(int, input().split())
s = input()

MOD = 998244353
KMAX = 26
kbits = 0
result = 0
dp = [0] * KMAX

if len(s) != n or n < 1 or n > m or m > 500000 or k < 1 or k > KMAX:
    sys.exit(1)

for c in s:
    ndp = [0] * KMAX
    ci = ord(c) - ord('a')
    if ci < 0 or ci >= KMAX:
        sys.exit(1)
    for j in range(ci):
        ndp[(kbits | (1 << j)).bit_count() - 1] += 1
    kbits |= (1 << ci)
    if kbits.bit_count() >= k:
        result += 1
    for j in range(1, KMAX):
        ndp[j - 1] += j * dp[j - 1]
        ndp[j] += (KMAX - j) * dp[j - 1]
    ndp[KMAX - 1] += KMAX * dp[KMAX - 1]
    for j in range(KMAX):
        ndp[j] %= MOD
    for j in range(k - 1, KMAX):
        result += ndp[j]
    result %= MOD
    dp = ndp

if kbits.bit_count() >= k:
    result -= 1
    result %= MOD

for _ in range(n, m):
    ndp = [0] * KMAX
    for j in range(1, KMAX):
        ndp[j - 1] += j * dp[j - 1]
        ndp[j] += (KMAX - j) * dp[j - 1]
    ndp[KMAX - 1] += KMAX * dp[KMAX - 1]
    for j in range(KMAX):
        ndp[j] %= MOD
    for j in range(k - 1, KMAX):
        result += ndp[j]
    result %= MOD
    dp = ndp

print(result)
0