Submission #60749184

---

Source Code Expand

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

#[allow(unused_imports)]
use proconio::{input, marker::Usize1};

fn main() {
    input!{
        h: usize,
        w: usize,
        x: u128,
        p: Usize1,
        q: Usize1,
        v: [[u128; w]; h],
    }

    let mut strng = v[p][q];

    let mut seen = vec![vec![false; w]; h];
    seen[p][q] = true;
    let mut pq = BinaryHeap::new();
    for r in 0..4 {
        let ni = p.wrapping_add(DI[r]);
        let nj = q.wrapping_add(DJ[r]);
        if ni < h && nj < w {
            seen[ni][nj] = true;
            pq.push((Reverse(v[ni][nj]), ni, nj));
        }
    }

    while let Some((Reverse(c), i, j)) = pq.pop() {
        if strng <= c * x {
            continue;
        }
        strng += c;
        for r in 0..4 {
            let ni = i.wrapping_add(DI[r]);
            let nj = j.wrapping_add(DJ[r]);
            if ni < h && nj < w && !seen[ni][nj] {
                seen[ni][nj] = true;
                pq.push((Reverse(v[ni][nj]), ni, nj));
            }
        }
    }

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

pub static INF: u64 = 1e18 as u64;
pub static DI: &[usize] = &[0, !0, 0, 1, !0, 1, !0, 1];
pub static DJ: &[usize] = &[!0, 0, 1, 0, !0, !0, 1, 1];
pub static DC: &[char] = &['L', 'U', 'R', 'D'];
pub trait Debuggable {
    fn debug_string(&self) -> String;
}

impl<T: std::fmt::Debug + std::fmt::Display> Debuggable for Vec<T> {
    fn debug_string(&self) -> String {
        use itertools::Itertools;
        use std::iter::repeat;
        if let Some(max_size) = self.iter()
            .map(|x| format!("{:?}", x).len())
            .max() {
                let mut idx = String::from("idx |");   
                let mut val = String::from("val |");   
                for (i, xi) in self.iter().enumerate() {
                    idx.push_str(
                        &format!(" {:>w$} |", i, w=max_size)
                    );
                    val.push_str(
                        &format!(" {:>w$} |", xi, w=max_size)
                    );
                }

                format!("{}\n{}\n{}\n", idx, repeat('-').take(idx.len()).join(""), val)
            } else {
                format!("idx | \nval |\n")
            }
    }
}

impl<T: std::fmt::Debug + std::fmt::Display> Debuggable for std::collections::BTreeSet<T> {
    fn debug_string(&self) -> String {
        use itertools::Itertools;
        format!("{{ {} }}", self.iter().join(", "))
    }
}

impl<T, U> Debuggable for std::collections::BTreeMap<T, U> 
where T: std::fmt::Debug + std::fmt::Display, U: std::fmt::Debug + std::fmt::Display
{
    fn debug_string(&self) -> String {
        use itertools::Itertools;
        format!(
            "{{\n{}\n }}", self.iter()
                .map(|(k, v)| format!("{k} -> {v}, "))
                .join("\n")
        )
    }
}

// lg! マクロの定義
#[macro_export]
macro_rules! lg {
    ($val:expr) => {
        #[cfg(feature = "local")]
        {
            {
                use Debuggable;
                let val = &$val;
                eprintln!(
                    "[{}:{}] {} = \n{}",
                    file!(),
                    line!(),
                    stringify!($val),
                    val.debug_string()
                );
                val
            }
        }
    }
}
    

Submission Info

Judge Result