結果

問題 No.1702 count good string
ユーザー phspls
提出日時 2022-10-02 01:26:07
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 1,810 bytes
コンパイル時間 13,243 ms
コンパイル使用メモリ 378,252 KB
実行使用メモリ 29,824 KB
最終ジャッジ日時 2024-12-24 07:38:53
合計ジャッジ時間 16,328 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 47
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: variable `LIMIT` should have a snake case name
  --> src/main.rs:12:9
   |
12 |     let LIMIT = 10;
   |         ^^^^^ help: convert the identifier to snake case: `limit`
   |
   = note: `#[warn(non_snake_case)]` on by default

ソースコード

diff #

const MOD: usize = 1e9 as usize + 7;

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 LIMIT = 10;
    let mut dp = vec![vec![vec![0usize; LIMIT]; 2]; n+1];
    dp[0][0][0] = 1;
    for i in 0..n {
        for pat in 0..2 {
            for j in 0..LIMIT {
                dp[i+1][pat][j] += dp[i][pat][j];
                dp[i+1][pat][j] %= MOD;
            }
            match s[i] {
                'y' => { dp[i+1][pat][1] += dp[i][pat][0]; dp[i+1][pat][1] %= MOD; },
                'u' => { dp[i+1][pat][2] += dp[i][pat][1]; dp[i+1][pat][2] %= MOD; },
                'k' => { dp[i+1][pat][3] += dp[i][pat][2]; dp[i+1][pat][3] %= MOD; },
                'i' => { dp[i+1][pat][4] += dp[i][pat][3]; dp[i+1][pat][4] %= MOD; },
                'c' => { dp[i+1][pat][5] += dp[i][pat][4]; dp[i+1][pat][5] %= MOD; },
                'o' => { dp[i+1][pat][6] += dp[i][pat][5]; dp[i+1][pat][6] %= MOD; },
                'd' => { dp[i+1][pat][7] += dp[i][pat][6]; dp[i+1][pat][7] %= MOD; },
                'e' => { dp[i+1][pat][8] += dp[i][pat][7]; dp[i+1][pat][8] %= MOD; },
                'r' => { dp[i+1][pat][9] += dp[i][pat][8]; dp[i+1][pat][9] %= MOD; },
                '?' => {
                    if pat == 0 {
                        for k in 0..LIMIT-1 {
                            dp[i+1][pat+1][k+1] += dp[i][pat][k];
                            dp[i+1][pat+1][k+1] %= MOD;
                        }
                    }
                },
                _ => {},
            };
        }
        
    }
    println!("{}", (dp[n][0][LIMIT-1] + dp[n][1][LIMIT-1]) % MOD);
}
0