結果

問題 No.2099 [Cherry Alpha B] Time Machine
コンテスト
ユーザー phspls
提出日時 2022-10-15 00:06:55
言語 Rust
(1.83.0 + proconio)
結果
TLE  
実行時間 -
コード長 1,694 bytes
コンパイル時間 18,100 ms
コンパイル使用メモリ 378,252 KB
実行使用メモリ 503,000 KB
最終ジャッジ日時 2024-06-26 18:16:07
合計ジャッジ時間 22,467 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 3 TLE * 3 -- * 66
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::{collections::{BinaryHeap, HashMap}, cmp::Reverse};

const INF: usize = 1usize << 60;

fn main() {
    let mut t = String::new();
    std::io::stdin().read_line(&mut t).ok();
    let t: isize = t.trim().parse().unwrap();
    let mut xa = String::new();
    std::io::stdin().read_line(&mut xa).ok();
    let xa: Vec<usize> = xa.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();
    let x = xa[0];
    let a = xa[1] as isize;
    let mut yb = String::new();
    std::io::stdin().read_line(&mut yb).ok();
    let yb: Vec<usize> = yb.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();
    let y = yb[0];
    let b = yb[1] as isize;

    let goal = t;
    let mut checked = HashMap::new();
    checked.insert(0isize, 0usize);
    let mut pque = BinaryHeap::new();
    pque.push((Reverse(0), 0isize));
    let get_cost = |p: isize, checked: &HashMap<isize, usize>| { *checked.get(&p).unwrap_or(&INF) };
    while let Some((Reverse(cost), pos)) = pque.pop() {
        if get_cost(goal, &checked) <= cost { break; }
        if get_cost(pos, &checked) < cost { continue; }
        let prev_cost = get_cost(pos+1, &checked);
        if pos+1 <= goal && prev_cost > cost + 1 {
            pque.push((Reverse(cost+1), pos+1));
            checked.insert(pos+1, cost+1);
        }
        if pos < goal && get_cost(pos+a, &checked) > cost + x {
            pque.push((Reverse(cost+x), pos+a));
            checked.insert(pos+a, cost+x);
        }
        if pos > goal && get_cost(pos-b, &checked) > cost + y {
            pque.push((Reverse(cost+y), pos-b));
            checked.insert(pos-b, cost+y);
        }
    }
    println!("{}", checked.get(&goal).unwrap());
}
0