結果

問題 No.2328 Build Walls
ユーザー zeronosu77108zeronosu77108
提出日時 2023-05-28 15:19:00
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 3,399 bytes
コンパイル時間 15,273 ms
コンパイル使用メモリ 379,540 KB
実行使用メモリ 53,220 KB
最終ジャッジ日時 2024-12-27 05:59:05
合計ジャッジ時間 27,111 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::collections::BTreeMap;

fn main() {
    let mut sc = Scanner::new();

    let h = sc.usize();
    let w = sc.usize();

    let mut a = vec![vec![0; w]; h-2];
    for i in 0..h-2 { a[i] = sc.vec::<i64>(w); }
    for i in 0..h-2 { for j in 0..w {
        if a[i][j] == -1 { a[i][j] = std::i32::MAX as i64; }
    }}

    let mut memo = BTreeMap::new();
    let mut ans = std::i32::MAX as i64;
    for i in 0..h-2 {
        ans = ans.min(f(i, 0, h, w, &a, &mut memo));
    }

    if ans >= std::i32::MAX as i64 { ans = -1; }
    println!("{}", ans);
}


fn f(i:usize, j:usize, h:usize, w:usize, a:&Vec<Vec<i64>>, memo:&mut BTreeMap<(usize,usize), i64>) -> i64 {
    if j == w { return 0; }
    if let Some(&x) = memo.get(&(i, j)) { return x; }

    let mut res = std::i32::MAX as i64;
    if i > 0 {
        res = res.min(f(i-1, j+1, h, w, a, memo) + a[i][j]);
        if memo.contains_key(&(i-1, j)) {
            res = res.min(f(i-1, j, h, w, a, memo) + a[i][j]);
        }
    }
    res = res.min(f(i, j+1, h, w, a, memo) + a[i][j]);
    if i+1 < h-2 {
        res = res.min(f(i+1, j+1, h, w, a, memo) + a[i][j]);
        res = res.min(f(i+1, j, h, w, a, memo) + a[i][j]);
    }

    memo.insert((i, j), res);
    res
}

struct Scanner { s : std::collections::VecDeque<String> } #[allow(unused)] impl Scanner { fn new() -> Self { use std::io::Read; let mut s = String::new(); std::io::stdin().read_to_string(&mut s).unwrap(); Self { s : s.split_whitespace().map(|s| s.to_string()).collect() } } fn reload(&mut self) -> () { use std::io::Read; let mut s = String::new(); std::io::stdin().read_to_string(&mut s).unwrap(); self.s = s.split_whitespace().map(|s| s.to_string()).collect(); } fn usize(&mut self) -> usize { self.input() } fn usize1(&mut self) -> usize { self.input::<usize>() - 1 } fn isize(&mut self) -> isize { self.input() } fn i32(&mut self) -> i32 { self.input() } fn i64(&mut self) -> i64 { self.input() } fn i128(&mut self) -> i128 { self.input() } fn u8(&mut self) -> u8 { self.input() } fn u32(&mut self) -> u32 { self.input() } fn u64(&mut self) -> u64 { self.input() } fn u128(&mut self) -> u128 { self.input() } fn edge(&mut self) -> (usize, usize) { (self.usize1(), self.usize1()) } fn edges(&mut self, m : usize) -> Vec<(usize, usize)> { let mut e = Vec::with_capacity(m); for _ in 0..m { e.push(self.edge()); } e } fn wedge<T:std::str::FromStr>(&mut self) -> (usize, usize, T) { (self.usize1(), self.usize1(), self.input()) } fn wedges<T:std::str::FromStr>(&mut self, m : usize) -> Vec<(usize, usize, T)> { let mut e = Vec::with_capacity(m); for _ in 0..m { e.push(self.wedge()); } e } fn input<T>(&mut self) -> T where T: std::str::FromStr { if self.s.is_empty() { self.reload(); } if let Some(head) = self.s.pop_front() { head.parse::<T>().ok().unwrap() } else { panic!() } } fn tuple<T, U>(&mut self) -> (T, U) where T: std::str::FromStr, U: std::str::FromStr { (self.input(), self.input()) } fn vec<T>(&mut self, n: usize) -> Vec<T> where T: std::str::FromStr { if self.s.is_empty() { self.reload(); } self.s.drain(..n).map(|s| s.parse::<T>().ok().unwrap() ).collect::<Vec<T>>() } fn nvec<T>(&mut self) -> Vec<T> where T: std::str::FromStr { let n : usize = self.input(); self.vec(n) } fn chars(&mut self) -> Vec<char> { let s : String = self.input(); s.chars().collect() } fn bytes(&mut self) -> Vec<u8> { let s : String = self.input(); s.bytes().collect() } }
0