結果

問題 No.1576 織姫と彦星
ユーザー elphe
提出日時 2026-02-13 13:25:57
言語 Rust
(1.93.0 + proconio + num + itertools)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 2,234 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,365 ms
コンパイル使用メモリ 206,884 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2026-02-13 13:26:01
合計ジャッジ時間 4,091 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 54
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

fn main() {
    let stdin = std::io::read_to_string(std::io::stdin().lock()).unwrap();
    let mut stdin = stdin.split_ascii_whitespace();

    let n: u32 = stdin.next().unwrap().parse().unwrap();
    let start: u32 = stdin.next().unwrap().parse().unwrap();
    let end: u32 = stdin.next().unwrap().parse().unwrap();
    let stones: Vec<u32> = (0..n)
        .map(|_| stdin.next().unwrap().parse().unwrap())
        .collect();

    use std::io::Write;
    std::io::stdout()
        .lock()
        .write_all(output(solve(start, end, stones)).as_bytes())
        .unwrap();
}

fn prepare(stones: Vec<u32>) -> Vec<Vec<u32>> {
    let mut groups = vec![Vec::new(); 31];
    stones
        .into_iter()
        .for_each(|s| groups[s.count_ones() as usize].push(s));
    groups
}

fn solve(start: u32, end: u32, stones: Vec<u32>) -> Option<u32> {
    let mut stones = prepare(stones);
    let mut q = std::collections::VecDeque::new();
    q.push_back((start, 0));
    while let Some((cur_pos, cur_dist)) = q.pop_front() {
        if (cur_pos ^ end).count_ones() == 1 {
            return Some(cur_dist);
        }
        if cur_pos.count_ones() >= 1 {
            stones[(cur_pos.count_ones() - 1) as usize] = stones[(cur_pos.count_ones() - 1) as usize]
                .iter()
                .filter_map(|&next| match (cur_pos ^ next).count_ones() == 1 {
                    true => {
                        q.push_back((next, cur_dist + 1));
                        None
                    }
                    false => Some(next),
                })
                .collect();
        }
        if cur_pos.count_ones() <= 29 {
            stones[(cur_pos.count_ones() + 1) as usize] = stones[(cur_pos.count_ones() + 1) as usize]
                .iter()
                .filter_map(|&next| match (cur_pos ^ next).count_ones() == 1 {
                    true => {
                        q.push_back((next, cur_dist + 1));
                        None
                    }
                    false => Some(next),
                })
                .collect();
        }
    }
    None
}

fn output(ans: Option<u32>) -> String {
    match ans {
        Some(ans) => ans.to_string(),
        None => String::from("-1"),
    }
}
0