結果

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

ソースコード

diff # 

use std::io::*;
use std::str::FromStr;
use utils::*;

pub fn main() {
    let i = stdin();
    let mut o = Vec::new();
    run(i.lock(), &mut o);
    stdout().write_all(&o).unwrap();
}

fn run<R: BufRead, W: Write>(i: R, o: &mut W) {
    let mut i = ReadEx::from(i);
    let n = i.read::<usize>();
    let r = solve(n);
    if r == usize::max_value() {
        writeln!(o, "-1")
    } else {
        writeln!(o, "{}", r)
    }.unwrap();
}
fn solve(n: usize) -> usize {
    fn bits(mut n: usize) -> usize {
        let mut c = 0;
        while n != 0 {
            if n % 2 == 1 {
                c += 1;
            }
            n /= 2;
        }
        c
    }
    let mut buf = vec![usize::max_value(); n + 1];
    let mut q = Vec::new();
    q.push(1);
    buf[0] = 0;
    buf[1] = 1;
    while let Some(i) = q.pop() {
        let mut update = |i0, i1| {
            let next = buf[i0] + 1;
            if buf[i1] > next {
                buf[i1] = next;
                q.push(i1);
            }
        };
        let b = bits(i);
        if b < i {
            update(i, i - b);
        }
        if i + b <= n {
            update(i, i + b);
        }
    }
    buf[n]
}

mod utils {
    use super::*;

    pub struct ReadEx<R: BufRead> {
        r: R,
        s: String,
    }

    impl<R: BufRead> ReadEx<R> {
        pub fn from(r: R) -> Self {
            ReadEx {
                r: r,
                s: String::new(),
            }
        }
        pub fn read_line(&mut self) -> &str {
            self.s.clear();
            self.r.read_line(&mut self.s).unwrap();
            self.s.trim()
        }

        pub fn read<T: FromStr>(&mut self) -> T {
            self.read_line().parse().ok().unwrap()
        }
    }
}
0