結果
問題 |
No.1646 Avoid Palindrome
|
ユーザー |
|
提出日時 | 2022-10-05 00:15:33 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 270 ms / 3,000 ms |
コード長 | 1,535 bytes |
コンパイル時間 | 18,530 ms |
コンパイル使用メモリ | 378,304 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-12-30 15:37:43 |
合計ジャッジ時間 | 28,813 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 40 |
ソースコード
const MOD: usize = 998244353; const SIZE: usize = 26; 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<_>>(); let mut dp = vec![vec![0usize; SIZE+1]; SIZE+1]; dp[SIZE][SIZE] = 1; for i in 0..n { let mut next_dp = vec![vec![0usize; SIZE+1]; SIZE+1]; if s[i] == '?' { let mut summary = vec![0usize; SIZE+1]; for pprev in 0..=SIZE { for prev in 0..=SIZE { summary[prev] += dp[pprev][prev]; summary[prev] %= MOD; } } for cur in 0..SIZE { for prev in 0..=SIZE { if prev == cur { continue; } next_dp[prev][cur] = MOD + summary[prev] - dp[cur][prev]; next_dp[prev][cur] %= MOD; } } } else { let idx = s[i] as usize - 'a' as usize; for prev in 0..=SIZE { if prev == idx { continue; } for pprev in 0..=SIZE { if pprev == idx { continue; } next_dp[prev][idx] += dp[pprev][prev]; next_dp[prev][idx] %= MOD; } } } dp = next_dp; } println!("{}", dp.iter().map(|v| v.iter().sum::<usize>()).sum::<usize>() % MOD); }