結果

問題 No.3 ビットすごろく
コンテスト
ユーザー Masaki Kitaguchi
提出日時 2022-01-11 06:08:14
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 56 ms / 5,000 ms
コード長 2,031 bytes
コンパイル時間 12,413 ms
コンパイル使用メモリ 401,176 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-14 11:24:01
合計ジャッジ時間 14,347 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#[macro_export]
macro_rules! setup {
    { mut $input:ident: SplitWhitespace $(,)? } => {
        use std::io::Read;
        let mut buf = String::new();
        std::io::stdin().read_to_string(&mut buf).unwrap();
        let mut $input = buf.split_whitespace();
    };
}

#[macro_export]
macro_rules! parse_next {
    ($str_iter:expr) => {
        $str_iter.next().unwrap().parse().unwrap()
    };
}

use std::collections::VecDeque;

fn main() {
    setup! { mut input: SplitWhitespace };

    let n: usize = parse_next!(input);

    macro_rules! ones {
        ($i:expr) => {
            format!("{:#b}", $i).replace("0b", "").matches("1").count()
        };
    }

    let nums = (1..=n).map(|i| i as usize).collect::<Vec<_>>();

    let ans = (|| {
        let mut queue = VecDeque::new();
        let mut visited = vec![false; n * 2];
        {
            queue.push_back(1);
            visited[0] = true;
        }
        let mut depth = 0;
        while !queue.is_empty() {
            depth += 1;
            let mut len = queue.len();
            while len > 0 {
                let i = queue.pop_front().unwrap();

                if i == n {
                    return depth;
                }

                let mut children = vec![];
                {
                    let j = i + ones!(i);
                    if nums.contains(&j) {
                        children.push(j);
                    }
                }
                {
                    if !(i <= ones!(i)) {
                        let j = i - ones!(i);
                        if nums.contains(&j) {
                            children.push(j);
                        }
                    }
                }
                for j in children.into_iter() {
                    if !visited[j - 1] {
                        queue.push_back(j);
                        visited[j - 1] = true;
                    }
                }
                len -= 1;
            }
        }
        return -1;
    })();

    println!("{:?}", ans);
}
0