結果
問題 |
No.1646 Avoid Palindrome
|
ユーザー |
|
提出日時 | 2022-10-03 23:33:03 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 314 ms / 3,000 ms |
コード長 | 1,549 bytes |
コンパイル時間 | 17,607 ms |
コンパイル使用メモリ | 379,296 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-12-28 04:45:33 |
合計ジャッジ時間 | 24,540 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 40 |
ソースコード
const MOD: usize = 998244353; const SIZE: usize = 27; fn main() { let mut n = String::new(); std::io::stdin().read_line(&mut n).ok(); let n: usize = n.trim().parse().unwrap(); let mut s = String::new(); std::io::stdin().read_line(&mut s).ok(); let s = s.trim().chars().collect::<Vec<char>>(); let mut dp = vec![vec![0usize; SIZE]; SIZE]; dp[26][26] = 1; for i in 0..n { let mut next_dp = vec![vec![0usize; SIZE]; SIZE]; if s[i] == '?' { let isum = (0..SIZE) .map(|prev| (0..SIZE).map(|pprev| dp[pprev][prev]).sum::<usize>() % MOD) .collect::<Vec<_>>(); next_dp = (0..SIZE).map(|prev| { (0..SIZE).map(|cur| { if cur == prev || cur == 26 { 0 } else { (MOD + isum[prev] - dp[cur][prev]) % MOD } }) .collect::<Vec<_>>() }) .collect::<Vec<_>>(); } else { let idx = s[i] as usize - 'a' as usize; for prev in 0..SIZE { if prev == idx { continue; } next_dp[prev][idx] = (0..SIZE) .filter(|&pprev| pprev != idx) .map(|pprev| dp[pprev][prev]) .sum::<usize>() % MOD; } } dp = next_dp; } println!("{}", dp.iter().map(|v| v.iter().sum::<usize>()).sum::<usize>() % MOD); }