結果

問題 No.1 道のショートカット
コンテスト
ユーザー qryxip
提出日時 2018-06-17 17:06:09
言語 Rust
(1.83.0 + proconio)
結果
MLE  
実行時間 -
コード長 4,124 bytes
コンパイル時間 14,283 ms
コンパイル使用メモリ 389,928 KB
実行使用メモリ 797,072 KB
最終ジャッジ日時 2024-07-08 05:02:53
合計ジャッジ時間 17,153 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 19 MLE * 1 -- * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#![allow(unused_imports)]
#![cfg_attr(feature = "cargo-clippy", allow(redundant_field_names))]

use std::collections::{BTreeMap, BTreeSet, BinaryHeap, HashMap, HashSet, VecDeque};
use std::io::{self, BufWriter, Read, Write};
use std::ops::{Add, Div, Mul, Sub};
use std::str::{self, FromStr};
use std::{cmp, fmt};

fn main() {
    let (n, c, ss, ts, ys, ms): (usize, u32, Vec<usize>, Vec<usize>, Vec<u32>, Vec<u32>) = {
        let mut sc = InputScanOnce::new(io::stdin(), 1024);
        let (n, c, v) = (sc.next(), sc.next(), sc.next());
        (n, c, sc.vec(v), sc.vec(v), sc.vec(v), sc.vec(v))
    };
    let g = {
        let mut g = vec![vec![]; n];
        for (((s, t), y), m) in ss.into_iter().zip(ts).zip(ys).zip(ms) {
            g[s - 1].push(E { t: t - 1, y, m });
        }
        g
    };
    let mut vs = vec![V { s: 0, y: 0, m: 0 }];
    let mut os: Vec<Option<O>> = vec![None; n];
    let mut r = None;
    while !vs.is_empty() {
        let mut vs_ = vec![];
        for v in vs {
            for e in &g[v.s] {
                let v = V {
                    s: e.t,
                    y: v.y + e.y,
                    m: v.m + e.m,
                };
                if v.y <= c {
                    if v.s == n - 1 {
                        r = Some(match r {
                            None => v.m,
                            Some(r) => cmp::min(r, v.m),
                        });
                    } else {
                        match os[v.s] {
                            Some(o) if o.y <= v.y && o.m <= v.m => {}
                            _ => {
                                os[v.s] = Some(O { y: v.y, m: v.m });
                                vs_.push(v)
                            }
                        }
                    }
                }
            }
        }
        vs = vs_;
    }
    match r {
        None => println!("-1"),
        Some(r) => println!("{}", r),
    }
}

#[derive(Clone, Copy)]
struct E {
    t: usize,
    y: u32,
    m: u32,
}

struct V {
    s: usize,
    y: u32,
    m: u32,
}

#[derive(Clone, Copy)]
struct O {
    y: u32,
    m: u32,
}

struct InputScanOnce {
    buf: Vec<u8>,
    pos: usize,
}

#[allow(dead_code)]
impl InputScanOnce {
    fn new<R: Read>(mut reader: R, estimated: usize) -> Self {
        let mut buf = Vec::with_capacity(estimated);
        let _ = io::copy(&mut reader, &mut buf).unwrap();
        InputScanOnce { buf: buf, pos: 0 }
    }

    #[inline]
    fn next<T: FromStr>(&mut self) -> T
    where
        T::Err: fmt::Debug,
    {
        let mut start = None;
        loop {
            match (self.buf[self.pos], start.is_some()) {
                (b' ', true) | (b'\n', true) => break,
                (_, true) | (b' ', false) | (b'\n', false) => self.pos += 1,
                (_, false) => start = Some(self.pos),
            }
        }
        let target = &self.buf[start.unwrap()..self.pos];
        unsafe { str::from_utf8_unchecked(target) }.parse().unwrap()
    }

    fn vec<T: FromStr>(&mut self, n: usize) -> Vec<T>
    where
        T::Err: fmt::Debug,
    {
        (0..n).map(|_| self.next()).collect()
    }

    fn pairs<T1: FromStr, T2: FromStr>(&mut self, n: usize) -> Vec<(T1, T2)>
    where
        T1::Err: fmt::Debug,
        T2::Err: fmt::Debug,
    {
        (0..n).map(|_| (self.next(), self.next())).collect()
    }

    fn trios<T1: FromStr, T2: FromStr, T3: FromStr>(&mut self, n: usize) -> Vec<(T1, T2, T3)>
    where
        T1::Err: fmt::Debug,
        T2::Err: fmt::Debug,
        T3::Err: fmt::Debug,
    {
        (0..n)
            .map(|_| (self.next(), self.next(), self.next()))
            .collect()
    }

    fn mat<T: FromStr>(&mut self, m: usize, n: usize) -> Vec<Vec<T>>
    where
        T::Err: fmt::Debug,
    {
        (0..m).map(|_| self.vec(n)).collect()
    }

    fn strings_as_mat<T, F: FnMut(u8) -> T>(&mut self, h: usize, mut f: F) -> Vec<Vec<T>> {
        (0..h)
            .map(|_| {
                let l = self.next::<String>();
                l.as_bytes().iter().cloned().map(&mut f).collect()
            })
            .collect()
    }
}
0