提出 #18833569

---

ソースコード 拡げる

use std::cmp::max;

fn main() {
    let (r, w) = (std::io::stdin(), std::io::stdout());
    let mut sc = IO::new(r.lock(), w.lock());

    let h: usize = sc.read();
    let w: usize = sc.read();
    let k: usize = sc.read();
    let mut map = vec![vec![0; w]; h];
    for _ in 0..k {
        let r = sc.read::<usize>() - 1;
        let c = sc.read::<usize>() - 1;
        let v = sc.read::<i64>();
        map[r][c] = v;
    }

    let mut dp = vec![vec![0; 4]; w];
    for i in 0..h {
        let mut next_dp = vec![vec![0; 4]; w];
        for j in 0..w {
            for cur in (0..3).rev() {
                dp[j][cur + 1] = max(dp[j][cur + 1], dp[j][cur] + map[i][j]);
            }
            for cur in 0..4 {
                if j + 1 < w {
                    dp[j + 1][cur] = max(dp[j + 1][cur], dp[j][cur]);
                }
                next_dp[j][0] = max(next_dp[j][0], dp[j][cur]);
            }
        }

        if i + 1 < h {
            dp = next_dp;
        }
    }
    println!("{}", *dp[w - 1].iter().max().unwrap());
}

pub struct IO<R, W: std::io::Write>(R, std::io::BufWriter<W>);

impl<R: std::io::Read, W: std::io::Write> IO<R, W> {
    pub fn new(r: R, w: W) -> Self {
        Self(r, std::io::BufWriter::new(w))
    }
    pub fn write<S: ToString>(&mut self, s: S) {
        use std::io::Write;
        self.1.write_all(s.to_string().as_bytes()).unwrap();
    }
    pub fn read<T: std::str::FromStr>(&mut self) -> T {
        use std::io::Read;
        let buf = self
            .0
            .by_ref()
            .bytes()
            .map(|b| b.unwrap())
            .skip_while(|&b| b == b' ' || b == b'\n' || b == b'\r' || b == b'\t')
            .take_while(|&b| b != b' ' && b != b'\n' && b != b'\r' && b != b'\t')
            .collect::<Vec<_>>();
        unsafe { std::str::from_utf8_unchecked(&buf) }
            .parse()
            .ok()
            .expect("Parse error.")
    }
    pub fn vec<T: std::str::FromStr>(&mut self, n: usize) -> Vec<T> {
        (0..n).map(|_| self.read()).collect()
    }
    pub fn chars(&mut self) -> Vec<char> {
        self.read::<String>().chars().collect()
    }
}

提出情報

ジャッジ結果