結果

問題 No.160 最短経路のうち辞書順最小
ユーザー lzy9lzy9
提出日時 2018-03-29 22:32:05
言語 Rust
(1.77.0 + proconio)
結果
AC  
実行時間 5 ms / 5,000 ms
コード長 2,092 bytes
コンパイル時間 15,197 ms
コンパイル使用メモリ 383,516 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-26 00:03:52
合計ジャッジ時間 13,276 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 1 ms
6,940 KB
testcase_03 AC 1 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 3 ms
6,940 KB
testcase_06 AC 3 ms
6,940 KB
testcase_07 AC 1 ms
6,944 KB
testcase_08 AC 1 ms
6,940 KB
testcase_09 AC 1 ms
6,944 KB
testcase_10 AC 1 ms
6,940 KB
testcase_11 AC 1 ms
6,940 KB
testcase_12 AC 2 ms
6,940 KB
testcase_13 AC 1 ms
6,944 KB
testcase_14 AC 2 ms
6,940 KB
testcase_15 AC 1 ms
6,940 KB
testcase_16 AC 1 ms
6,940 KB
testcase_17 AC 1 ms
6,940 KB
testcase_18 AC 1 ms
6,940 KB
testcase_19 AC 2 ms
6,940 KB
testcase_20 AC 1 ms
6,944 KB
testcase_21 AC 1 ms
6,940 KB
testcase_22 AC 2 ms
6,944 KB
testcase_23 AC 2 ms
6,944 KB
testcase_24 AC 1 ms
6,944 KB
testcase_25 AC 2 ms
6,944 KB
testcase_26 AC 1 ms
6,940 KB
testcase_27 AC 1 ms
6,940 KB
testcase_28 AC 5 ms
6,940 KB
testcase_29 AC 1 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io;
use std::cmp;
use std::cmp::Ordering;
use std::usize;
use std::collections::BinaryHeap;

fn main() {
    solve();
}

fn solve() {
    let mut str = String::new();
    io::stdin().read_line(&mut str).unwrap();
    let mut input: Vec<usize> = str.split_whitespace().map(|x| x.parse().unwrap()).collect();
    let n = input[0];
    let m = input[1];
    let s = input[2];
    let g = input[3];

    let mut graph: Vec<Vec<Edge>> = (0..n).map(|_| Vec::new()).collect();

    for _ in 0..m {
        str = String::new();
        io::stdin().read_line(&mut str).unwrap();
        input = str.split_whitespace().map(|x| x.parse().unwrap()).collect();
        graph[input[0]].push(Edge { to: input[1], cost: input[2] });
        graph[input[1]].push(Edge { to: input[0], cost: input[2] });
    }

    let mut dist: Vec<_> = (0..n).map(|_| usize::MAX).collect();
    let mut pre: Vec<_> = (0..n).map(|i| i).collect();
    let mut heap = BinaryHeap::new();

    heap.push(Node { id: g, cost: 0 });
    dist[g] = 0;

    while let Some(Node {id, cost}) = heap.pop() {
        if dist[id] < cost {
            continue;
        }

        for edge in &graph[id] {
            if cost + edge.cost < dist[edge.to] {
                heap.push(Node { id: edge.to, cost: cost + edge.cost });
                dist[edge.to] = cost + edge.cost;
                pre[edge.to] = id;
            }

            if cost + edge.cost == dist[edge.to] {
                pre[edge.to] = cmp::min(pre[edge.to], id);
            }
        }
    }

    let mut current = s;

    while current != g {
        print!("{} ", current);
        current = pre[current];
    }

    println!("{}", current);
}

struct Edge {
    to: usize,
    cost: usize,
}

#[derive(Copy, Clone, Eq, PartialEq)]
struct Node {
    id: usize,
    cost: usize,
}

impl Ord for Node {
    fn cmp(&self, other: &Self) -> Ordering {
        other.cost.cmp(&self.cost).then_with(|| self.id.cmp(&other.id))
    }
}

impl PartialOrd for Node {
    fn partial_cmp(&self, other: &Node) -> Option<Ordering> {
        Some(self.cmp(other))
    }
}
0