結果

問題 No.533 Mysterious Stairs
ユーザー Seiya
提出日時 2017-07-23 16:48:21
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 79 ms / 5,000 ms
コード長 1,030 bytes
コンパイル時間 13,629 ms
コンパイル使用メモリ 407,460 KB
実行使用メモリ 56,424 KB
最終ジャッジ日時 2024-10-09 11:00:23
合計ジャッジ時間 15,132 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::stdin;

fn main(){
    let n: usize = getvec()[0];
    println!("{}", solve(n));
}

fn solve(n: usize) -> usize {
    let mut dp: Vec<Vec<_>> = Vec::new();
    dp.push(vec![1, 1, 1]);
    dp.push(vec![1, 0, 0]);
    dp.push(vec![0, 1, 0]);
    dp.push(vec![0, 0, 1]);
    for i in 1..n {
        dp[i+1][0] = (dp[i+1][0]+dp[i][1]+dp[i][2]) % 1000000007;
        dp[i+2][1] = (dp[i+2][1]+dp[i][0]+dp[i][2]) % 1000000007;
        let x = (dp[i][0]+dp[i][1]) % 1000000007;
        dp.push(vec![0, 0, x]);
    }
    dp[n].iter().fold(0, |acc, x| acc+x) % 1000000007
}

#[allow(dead_code)]
fn getline() -> String {
    let mut s = String::new();
    match stdin().read_line(&mut s){
        Ok(_) => {s.trim().to_string()}
        Err(_) => String::new()
    }
}

#[allow(dead_code)]
fn getvec<T: std::str::FromStr>() -> Vec<T>{
    let line = getline();
    line.split_whitespace().filter_map(|x| x.parse().ok()).collect()
}

#[allow(dead_code)]
fn getpair() -> (usize, usize){
    let nm = getvec();
    (nm[0], nm[1])
}
0