結果

問題 No.3 ビットすごろく
ユーザー iwotiwot
提出日時 2019-08-14 15:00:23
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 73 ms / 5,000 ms
コード長 1,765 bytes
コンパイル時間 12,919 ms
コンパイル使用メモリ 389,812 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-01 09:28:14
合計ジャッジ時間 14,837 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

fn read<T: std::str::FromStr>() -> T {
    let mut s = String::new();
    std::io::stdin().read_line(&mut s).ok();
    s.trim().parse().ok().unwrap()
}

fn main() {
    let n:u32 = read();

    #[derive(Debug, Clone)]
    struct Node {
        to: Vec<usize>,
        done: bool,
        cost: i32,
    }

    let mut nodes: Vec<Node> = vec![];
    for i in 1..=n {
        let prev_idx = i as i32 - 1 - i.count_ones() as i32;
        let next_idx = i as i32 - 1 + i.count_ones() as i32;
        let done = false;
        let cost = -1;
        let mut to = vec![];
        if prev_idx >= 0 {
            to.push(prev_idx as usize);
        }
        if next_idx < n as i32 {
            to.push(next_idx as usize);
        }
        nodes.push(Node{to, done, cost});
    }

    nodes[0].cost = 0;

    loop {
        let mut done_node_idx: Option<usize> = None;
        let max = nodes.len();
        for i in 0..max {
            if nodes[i].done || nodes[i].cost < 0 {
                continue;
            }
            if done_node_idx.is_none() || nodes[i].cost < nodes[done_node_idx.unwrap()].cost {
                done_node_idx = Some(i);
            }
        }

        if done_node_idx.is_none() {
            break;
        }

        // let done_node = &mut nodes[done_node_idx.unwrap()];
        nodes[done_node_idx.unwrap()].done = true;

        for toidx in nodes[done_node_idx.unwrap()].to.clone() {
            let cost = nodes[done_node_idx.unwrap()].cost + 1;
            if nodes[toidx].cost < 0 || cost < nodes[toidx].cost {
                nodes[toidx].cost = cost;
            }
        }
    }

    if nodes[nodes.len()-1].cost >= 0 {
        println!("{:?}", nodes[nodes.len()-1].cost + 1);
    } else {
        println!("-1");
    }
}
0