結果

問題 No.2388 At Least K-Characters
コンテスト
ユーザー 👑 Mizar
提出日時 2023-07-12 22:12:50
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 161 ms / 4,000 ms
コード長 1,018 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,144 ms
コンパイル使用メモリ 85,112 KB
実行使用メモリ 82,048 KB
最終ジャッジ日時 2026-03-26 10:40:42
合計ジャッジ時間 5,763 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge3_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys

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

MOD = 998244353
KMAX = 26
kbits = 0
result = 0
dp = [0] * (KMAX + 2)

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

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

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

print(result)
0