#891938 (C++23) No.2388 At Least K-Characters

• 問題: No.2388 At Least K-Characters
• ユーザー: 👑 Mizar
• 提出日時: 2023-07-15 20:53:55
• 言語: C++23 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 51 ms / 4,000 ms
• コード長: 1,487 bytes
• コンパイル時間: 1,030 ms
• コンパイル使用メモリ: 87,548 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-05 03:38:55
• 合計ジャッジ時間: 3,231 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	2 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	2 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	2 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	49 ms; 5,376 KB
testcase_17	AC	49 ms; 5,376 KB
testcase_18	AC	50 ms; 5,376 KB
testcase_19	AC	50 ms; 5,376 KB
testcase_20	AC	50 ms; 5,376 KB
testcase_21	AC	48 ms; 5,376 KB
testcase_22	AC	47 ms; 5,376 KB
testcase_23	AC	49 ms; 5,376 KB
testcase_24	AC	49 ms; 5,376 KB
testcase_25	AC	48 ms; 5,376 KB
testcase_26	AC	50 ms; 5,376 KB
testcase_27	AC	50 ms; 5,376 KB
testcase_28	AC	49 ms; 5,376 KB
testcase_29	AC	51 ms; 5,376 KB
testcase_30	AC	50 ms; 5,376 KB
testcase_31	AC	50 ms; 5,376 KB
testcase_32	AC	50 ms; 5,376 KB
testcase_33	AC	50 ms; 5,376 KB
testcase_34	AC	49 ms; 5,376 KB
testcase_35	AC	50 ms; 5,376 KB
testcase_36	AC	50 ms; 5,376 KB

ソースコード

#pragma GCC optimize ("O3")
#pragma GCC target ("sse,sse2,sse3,ssse3,sse4,popcnt,abm,avx")

#include <array> // array, swap
#include <bit> // (C++20) popcount
#include <cstdlib> // abort
#include <iostream> // cin, cout, endl
#include <string> // string

#define MOD 998244353
#define KMAX 26
#define KMAX2 (KMAX+2)

using ull = unsigned long long;

int main() {
	ull n, m, k;
	std::string s;
	std::cin >> n >> m >> k >> s;
	
    if(n < 1 || n > m || m > 500000 || s.length() != n || k < 1 || k > KMAX) {
        std::abort();
    }

	ull kbits = 0, kbc = 0, result = 0;
	std::array<ull, KMAX2> dp, ndp;
	dp.fill(0);

	for(auto it = s.begin(); it != s.end(); it++) {
		ull ci = *it - 'a';
		if(ci >= KMAX) {
		    std::abort;
		}
		ndp.fill(0);
		ull same = std::popcount(kbits & ((1 << ci) - 1));
		ndp[kbc] += same;
		ndp[kbc + 1] += ci - same;
		if((kbits & (1 << ci)) == 0) {
			kbits |= (1 << ci);
			kbc += 1;
		}
		if(kbc >= k) {
			result += 1;
		}
		for(ull j = 1; j <= KMAX; j++) {
			ndp[j] += j * dp[j];
			ndp[j + 1] += (KMAX - j) * dp[j];
			ndp[j] %= MOD;
		}
		for(ull j = k; j <= KMAX; j++) {
			result += ndp[j];
		}
		dp.swap(ndp);
	}

	if(kbc >= k) {
		result -= 1;
	}

	for(ull i = n; i < m; i++) {
		ndp.fill(0);
		for(ull j = 1; j <= KMAX; j++) {
			ndp[j] += j * dp[j];
			ndp[j + 1] += (KMAX - j) * dp[j];
			ndp[j] %= MOD;
		}
		for(ull j = k; j <= KMAX; j++) {
			result += ndp[j];
		}
		dp.swap(ndp);
	}
	
	result %= MOD;
	std::cout << result << std::endl;
}