結果

問題 No.762 PDCAパス
ユーザー phspls
提出日時 2020-07-12 23:05:21
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 60 ms / 2,000 ms
コード長 1,443 bytes
コンパイル時間 13,246 ms
コンパイル使用メモリ 397,884 KB
実行使用メモリ 12,672 KB
最終ジャッジ日時 2024-11-06 06:48:49
合計ジャッジ時間 16,871 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

const DIVISOR: usize = 1_000_000_007;
const NAME: &str = "PDCA";

fn main() {
    let mut nm = String::new();
    std::io::stdin().read_line(&mut nm).ok();
    let nm: Vec<usize> = nm.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();
    let n = nm[0];
    let m = nm[1];
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    let s: Vec<char> = s.trim().chars().collect();
    let mut dp: Vec<Vec<usize>> = vec![vec![0; n]; 4];
    let mut path: Vec<Vec<usize>> = vec![vec![]; n];
    for _ in 0..m {
        let mut uv = String::new();
        std::io::stdin().read_line(&mut uv).ok();
        let uv: Vec<usize> = uv.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();
        let u = uv[0] - 1;
        let v = uv[1] - 1;
        path[u].push(v);
        path[v].push(u);
    }

    for i in 0..4 {
        let charactor: char = NAME.chars().nth(3-i).unwrap();
        for j in 0..n {
            if i == 0 {
                dp[i][j] = if s[j] == charactor { 1 } else { 0 };
            } else {
                if dp[i-1][j] > 0 {
                    path[j].iter()
                        .filter(|&node| s[*node] == charactor)
                        .for_each(|&node| {
                            dp[i][node] += dp[i-1][j];
                        });
                }
            }
        }
    }
    println!("{}", dp[3].iter().map(|i| i % DIVISOR).sum::<usize>() % DIVISOR);
}
0