結果
問題 | No.20 砂漠のオアシス |
ユーザー | phspls |
提出日時 | 2022-09-03 02:23:12 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 15 ms / 5,000 ms |
コード長 | 2,018 bytes |
コンパイル時間 | 12,073 ms |
コンパイル使用メモリ | 380,880 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-11-16 10:11:19 |
合計ジャッジ時間 | 13,253 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 21 |
ソースコード
use std::collections::BinaryHeap;const DX: [isize; 4] = [1,-1,0,0];const DY: [isize; 4] = [0,0,1,-1];fn main() {let mut nvxy = String::new();std::io::stdin().read_line(&mut nvxy).ok();let nvxy: Vec<usize> = nvxy.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();let n = nvxy[0];let v = nvxy[1];let ox = nvxy[3];let oy = nvxy[2];let ox = if ox > 0 { ox - 1 } else { ox };let oy = if oy > 0 { oy - 1 } else { oy };let unused = ox == 0 && oy == 0;let grid = (0..n).map(|_| {let mut temp = String::new();std::io::stdin().read_line(&mut temp).ok();let temp: Vec<usize> = temp.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();temp}).collect::<Vec<Vec<usize>>>();let mut result = vec![vec![vec![0usize; 2]; n]; n];let mut pque = BinaryHeap::new();pque.push((v, 0, 0, 0));while let Some((vit, x, y, drunk)) = pque.pop() {if vit < result[x][y][drunk] { continue; }for direct in 0..4 {let nx = x as isize + DX[direct];let ny = y as isize + DY[direct];if nx >= 0 && ny >= 0 && nx < n as isize && ny < n as isize {let nx = nx as usize;let ny = ny as usize;if grid[nx][ny] >= vit { continue; }let nvit = vit - grid[nx][ny];let ndrunk = drunk.max(if !unused && nx == ox && ny == oy { 1 } else { 0 });let nvit = if drunk == 0 && !unused && nx == ox && ny == oy { nvit * 2 } else { nvit };if nx == n-1 && ny == n-1 {result[nx][ny][ndrunk] = nvit;println!("YES");return;}if result[nx][ny][ndrunk] < nvit {result[nx][ny][ndrunk] = nvit;pque.push((nvit, nx, ny, ndrunk));}}}}println!("NO");}