結果

問題 No.2388 At Least K-Characters
ユーザー 👑 MizarMizar
提出日時 2023-07-15 13:23:03
言語 Rust
(1.77.0)
結果
AC  
実行時間 33 ms / 4,000 ms
コード長 1,775 bytes
コンパイル時間 662 ms
コンパイル使用メモリ 140,944 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-09-18 12:51:00
合計ジャッジ時間 2,819 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 1 ms
4,380 KB
testcase_03 AC 1 ms
4,380 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 1 ms
4,376 KB
testcase_06 AC 1 ms
4,380 KB
testcase_07 AC 1 ms
4,376 KB
testcase_08 AC 1 ms
4,380 KB
testcase_09 AC 1 ms
4,380 KB
testcase_10 AC 1 ms
4,376 KB
testcase_11 AC 1 ms
4,376 KB
testcase_12 AC 1 ms
4,376 KB
testcase_13 AC 1 ms
4,380 KB
testcase_14 AC 1 ms
4,376 KB
testcase_15 AC 1 ms
4,376 KB
testcase_16 AC 32 ms
4,376 KB
testcase_17 AC 32 ms
4,380 KB
testcase_18 AC 31 ms
4,376 KB
testcase_19 AC 31 ms
4,376 KB
testcase_20 AC 32 ms
4,376 KB
testcase_21 AC 33 ms
4,380 KB
testcase_22 AC 33 ms
4,376 KB
testcase_23 AC 32 ms
4,376 KB
testcase_24 AC 31 ms
4,376 KB
testcase_25 AC 31 ms
4,380 KB
testcase_26 AC 32 ms
4,376 KB
testcase_27 AC 32 ms
4,376 KB
testcase_28 AC 33 ms
4,380 KB
testcase_29 AC 33 ms
4,376 KB
testcase_30 AC 32 ms
4,380 KB
testcase_31 AC 32 ms
4,380 KB
testcase_32 AC 32 ms
4,376 KB
testcase_33 AC 31 ms
4,380 KB
testcase_34 AC 33 ms
4,380 KB
testcase_35 AC 32 ms
4,380 KB
testcase_36 AC 32 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::prelude::*;
fn main() {
    let mut input = String::new();
    std::io::stdin().read_to_string(&mut input).unwrap();
    const MOD: u64 = 998244353;
    const KMAX: usize = 26;
    const KMAX2: usize = KMAX + 2;
    let mut tokens = input.split_ascii_whitespace();
    let n = tokens.next().unwrap().parse::<usize>().unwrap();
    let m = tokens.next().unwrap().parse::<usize>().unwrap();
    let k = tokens.next().unwrap().parse::<usize>().unwrap();
    let s = tokens.next().unwrap();
    if n < 1 || n != s.len() || n > m || m > 500000 || k < 1 || k > KMAX {
        panic!();
    }
    let mut kbits = 0u32;
    let mut kbc = 0usize;
    let mut dp = [0u64; KMAX2];
    let mut result = 0u64;
    for c in s.bytes() {
        let mut ndp = [0u64; KMAX2];
        let ci = (c - b'a') as usize;
        if ci >= KMAX {
            panic!();
        }
        let same = (kbits & ((1 << ci) - 1)).count_ones() as u64;
		ndp[kbc] += same;
		ndp[kbc + 1] += (ci as u64) - same;
		if (kbits & (1 << ci)) == 0 {
			kbits |= 1 << ci;
			kbc += 1;
		}
		if kbc >= k {
			result += 1;
		}
        for j in 1..=KMAX {
            ndp[j] += (j as u64) * dp[j];
            ndp[j + 1] += ((KMAX - j) as u64) * dp[j];
            ndp[j] %= MOD;
        }
        for &je in ndp[k..=KMAX].iter() {
            result += je;
        }
        dp = ndp;
    }
    if kbc >= k {
        result -= 1;
    }
    for _ in n..m {
        let mut ndp = [0u64; KMAX2];
        for j in 1..=KMAX {
            ndp[j] += (j as u64) * dp[j];
            ndp[j + 1] += ((KMAX - j) as u64) * dp[j];
            ndp[j] %= MOD;
        }
        for &je in ndp[k..=KMAX].iter() {
            result += je;
        }
        dp = ndp;
    }
    result %= MOD;
    println!("{}", result);
}
0