結果

問題 No.3 ビットすごろく
ユーザー TiramisterTiramister
提出日時 2018-09-03 17:56:05
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,799 bytes
コンパイル時間 12,441 ms
コンパイル使用メモリ 391,484 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-01 09:06:22
合計ジャッジ時間 13,733 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::*;
use std::str::FromStr;
use std::collections::VecDeque;

#[allow(dead_code)]
fn get_line() -> String {
    let stdin = stdin();
    let mut line = String::new();
    stdin.lock().read_line(&mut line).expect("io error.");
    line.trim().to_string()
}

#[allow(dead_code)]
fn cast<T: FromStr>(s: &str) -> T {
    s.parse().ok().expect("parse error.")
}

#[allow(dead_code)]
fn get_vec<T: FromStr>() -> Vec<T> {
    (&get_line()).split(' ').map(cast::<T>).collect()
}

fn main() {
    let n: usize = cast(&get_line());

    let mut d: Vec<Option<u64>> = vec![None; n + 1];
    // 移動数をメモする配列
    // None    => 未探索
    // Some(x) => 移動数xで到達可能

    // 以降は幅優先探索
    let mut que = VecDeque::<usize>::new();
    que.push_back(1);
    d[1] = Some(1);

    // d[v]を更新するための関数
    let update = |v: usize,
                  dis: u64,
                  que: &mut VecDeque<usize>,
                  d: &mut Vec<Option<u64>>| {
        match d[v] {
            // vが探索済みならスルー
            Some(_) => (),

            // そうでなければdis+1で更新し、queueに突っ込む
            None => {
                d[v] = Some(dis + 1);
                que.push_back(v);
            },
        };
    };

    while !que.is_empty() {
        let v = que.pop_front().unwrap();
        let dis = d[v].unwrap();

        // vの立っているbitの数(bitcount)
        let w = (v as u64).count_ones() as usize;

        if v + w <= n {
            update(v + w, dis, &mut que, &mut d);
        };
        if v > w {
            update(v - w, dis, &mut que, &mut d);
        };
    }

    println!("{}",
        match d[n] {
            Some(k) => k as i64,
            None => -1,
        }
    );
}
0