Problem:
Score : 500 points

Problem Statement

There are K items placed on a grid of squares with R rows and C columns. Let (i, j) denote the square at the i-th row (1 \leq i \leq R) and the j-th column (1 \leq j \leq C). The i-th item is at (r_i, c_i) and has the value v_i.

Takahashi will begin at (1, 1), the start, and get to (R, C), the goal. When he is at (i, j), he can move to (i + 1, j) or (i, j + 1) (but cannot move to a non-existent square).

He can pick up items on the squares he visits, including the start and the goal, but at most three for each row. It is allowed to ignore the item on a square he visits.

Find the maximum possible sum of the values of items he picks up.

Constraints

1 \leq R, C \leq 3000

1 \leq K \leq \min(2 \times 10^5, R \times C)

1 \leq r_i \leq R

1 \leq c_i \leq C

(r_i, c_i) \neq (r_j, c_j) (i \neq j)

1 \leq v_i \leq 10^9

All values in input are integers.

Input

Input is given from Standard Input in the following format:

R C K
r_1 c_1 v_1
r_2 c_2 v_2
:
r_K c_K v_K

Output

Print the maximum possible sum of the values of items Takahashi picks up.

Sample Input 1

2 2 3
1 1 3
2 1 4
1 2 5

Sample Output 1

8

He has two ways to get to the goal:

Visit (1, 1), (1, 2), and (2, 2), in this order. In this case, the total value of the items he can pick up is 3 + 5 = 8.

Visit (1, 1), (2, 1), and (2, 2), in this order. In this case, the total value of the items he can pick up is 3 + 4 = 7.

Thus, the maximum possible sum of the values of items he picks up is 8.

Sample Input 2

2 5 5
1 1 3
2 4 20
1 2 1
1 3 4
1 4 2

Sample Output 2

29

We have four items in the 1-st row. The optimal choices are as follows:

Visit (1, 1) (1, 2), (1, 3), (1, 4), (2, 4), and (2, 5), in this order, and pick up all items except the one on (1, 2). Then, the total value of the items he picks up will be 3 + 4 + 2 + 20 = 29.

Sample Input 3

4 5 10
2 5 12
1 5 12
2 3 15
1 2 20
1 1 28
2 4 26
3 2 27
4 5 21
3 5 10
1 3 10

Sample Output 3

142

Solution in Rust (1.42.0) (judge verdict: Runtime Error)

// -*- coding:utf-8-unix -*-

#[macro_use]
extern crate lazy_static;
extern crate num_bigint; // 0.2.2
extern crate num_traits; // 0.2.8
use num_bigint::BigInt;
use num_traits::Pow;

// use proconio::derive_readable;
use proconio::fastout;
use proconio::input;
// use std::convert::TryInto;
use libm::*;
use std::cmp::*;
use std::collections::{BinaryHeap, HashMap, HashSet};
use std::io::*;
use std::ops::Range;
use std::str::FromStr;
use superslice::*;

pub fn read<T: FromStr>() -> T {
    let stdin = stdin();
    let stdin = stdin.lock();
    let token: String = stdin
        .bytes()
        .map(|c| c.expect("failed to read char") as char)
        .skip_while(|c| c.is_whitespace())
        .take_while(|c| !c.is_whitespace())
        .collect();
    token.parse().ok().expect("failed to parse token")
}

//abc175-E
// #[fastout]
fn main() {
    input![
        r: usize,
        c: usize,
        k: usize,
        cell: [(usize, usize, usize); k]
    ];

    // eprintln!("{}", r);
    // eprintln!("{}", r);
    // eprintln!("{}", k);
    // eprintln!("{:?}", cell);

    // let mut cs = vec![Vec::new(); r];
    // let mut vs = vec![Vec::new(); r];
    let mut items = HashMap::new();
    // let mut com_count = vec![vec![0; c]; r];

    for (ri, ci, vi) in cell {
        // cs[ri - 1].push(ci - 1);
        // vs[ri - 1].push(vi - 1);
        items.insert((ri - 1, ci - 1), vi);
    }

    // for i in 0..r {
    //     cs[i].sort();
    //     vs[i].sort();
    // }

    // for ri in 0..r {
    //     for ci in 0..c {
    //         if let Some(_) = items.get(&(ri, ci)) {
    //             com_count[ri][ci] = if ci == 0 || ci == c - 1 {
    //                 1
    //             } else {
    //                 com_count[ri][ci - 1] + 1
    //             }
    //         }
    //     }
    // }

    let mut anss: Vec<Vec<Vec<i64>>> = vec![vec![vec![-1; 4]; c]; r];

    let goal_item_v = items.get(&(r - 1, c - 1)).unwrap_or(&0);
    anss[r - 1][c - 1][0] = 0 + *goal_item_v as i64;
    anss[r - 1][c - 1][1] = 0 + *goal_item_v as i64;
    anss[r - 1][c - 1][2] = 0 + *goal_item_v as i64;
    anss[r - 1][c - 1][3] = 0;

    for delta_c in 0..c - 1 {
        let delta_c = (c - 2) - delta_c;
        for a in 0..3 {
            anss[r - 1][delta_c][a] = if let Some(v) = items.get(&(r - 1, delta_c)) {
                max(
                    anss[r - 1][delta_c + 1][a + 1] + *v as i64,
                    anss[r - 1][delta_c + 1][a],
                )
            } else {
                anss[r - 1][delta_c + 1][a]
            };
        }

        anss[r - 1][delta_c][3] = 0;

        // eprintln!("[{},{}][0] = {}", r - 1, delta_c, anss[r - 1][delta_c][0]);
        // eprintln!("[{},{}][1] = {}", r - 1, delta_c, anss[r - 1][delta_c][1]);
        // eprintln!("[{},{}][2] = {}", r - 1, delta_c, anss[r - 1][delta_c][2]);
        // eprintln!("[{},{}][3] = {}", r - 1, delta_c, anss[r - 1][delta_c][3]);
    }
    for delta_r in 0..r - 1 {
        let delta_r = r - 2 - delta_r;
        for a in 0..4 {
            anss[delta_r][c - 1][a] = if let Some(v) = items.get(&(delta_r, c - 1)) {
                anss[delta_r + 1][c - 1][0] + *v as i64
            } else {
                anss[delta_r + 1][c - 1][0]
            }
        }

        // eprintln!("[{},{}][0] = {}", delta_r, c - 1, anss[delta_r][c - 1][0]);
        // eprintln!("[{},{}][1] = {}", delta_r, c - 1, anss[delta_r][c - 1][1]);
        // eprintln!("[{},{}][2] = {}", delta_r, c - 1, anss[delta_r][c - 1][2]);
        // eprintln!("[{},{}][3] = {}", delta_r, c - 1, anss[delta_r][c - 1][3]);
    }

    // eprintln!("r,c = {},{}", r, c);
    for l_s in 1..max(r, c) {
        let r_idx = max((r - 1) as i64 - l_s as i64, 0) as usize;
        let c_idx = max((c - 1) as i64 - l_s as i64, 0) as usize;
        // eprintln!("r_idx: {}, c_idx: {}", r_idx, c_idx);

        for delta_c in (c_idx + 1)..(c - 1) {
            let delta_c = (c_idx + 1) + (c - 2) - delta_c;
            // eprintln!("(r_idx, delta_c)={}, {}", r_idx, delta_c);
            anss[r_idx][delta_c][3] = anss[r_idx + 1][delta_c][0];
            //delta_c ~ cまでにあとどれくらいアイテムがあるか？
            if let Some(v) = items.get(&(r_idx, delta_c)) {
                for item_count in 0..3 {
                    anss[r_idx][delta_c][item_count] = *[
                        anss[r_idx][delta_c + 1][item_count],
                        anss[r_idx][delta_c + 1][item_count + 1] + *v as i64,
                        anss[r_idx + 1][delta_c][0] + *v as i64,
                    ]
                    .iter()
                    .max()
                    .unwrap();
                }
            } else {
                for item_count in 0..3 {
                    anss[r_idx][delta_c][item_count] = *[
                        anss[r_idx][delta_c + 1][item_count],
                        anss[r_idx + 1][delta_c][0],
                    ]
                    .iter()
                    .max()
                    .unwrap();
                }
            }

            // eprintln!("[{},{}][0] = {}", r_idx, delta_c, anss[r_idx][delta_c][0]);
            // eprintln!("[{},{}][1] = {}", r_idx, delta_c, anss[r_idx][delta_c][1]);
            // eprintln!("[{},{}][2] = {}", r_idx, delta_c, anss[r_idx][delta_c][2]);
            // eprintln!("[{},{}][3] = {}", r_idx, delta_c, anss[r_idx][delta_c][3]);
        }

        for delta_r in r_idx..r - 1 {
            let delta_r = r_idx + r - 2 - delta_r;
            // eprintln!("(delta_r, c_idx)={}, {}", delta_r, c_idx);

            anss[delta_r][c_idx][3] = anss[delta_r + 1][c_idx][0];
            //delta_c ~ cまでにあとどれくらいアイテムがあるか？
            if let Some(v) = items.get(&(delta_r, c_idx)) {
                for item_count in 0..3 {
                    anss[delta_r][c_idx][item_count] = *[
                        anss[delta_r][c_idx + 1][item_count],
                        anss[delta_r][c_idx + 1][item_count + 1] + *v as i64,
                        anss[delta_r + 1][c_idx][0] + *v as i64,
                    ]
                    .iter()
                    .max()
                    .unwrap();
                }
            } else {
                for item_count in 0..3 {
                    anss[delta_r][c_idx][item_count] = *[
                        anss[delta_r][c_idx + 1][item_count],
                        anss[delta_r + 1][c_idx][0],
                    ]
                    .iter()
                    .max()
                    .unwrap();
                }
            }

            // eprintln!("[{},{}][0] = {}", delta_r, c_idx, anss[delta_r][c_idx][0]);
            // eprintln!("[{},{}][1] = {}", delta_r, c_idx, anss[delta_r][c_idx][1]);
            // eprintln!("[{},{}][2] = {}", delta_r, c_idx, anss[delta_r][c_idx][2]);
            // eprintln!("[{},{}][3] = {}", delta_r, c_idx, anss[delta_r][c_idx][3]);
        }
    }

    println!("{}", anss[0][0][0]);
}

fn go(mid: usize, d: usize, use_num: usize, num753_cand: &mut Vec<usize>, n: usize) {}

// use lazy_static::lazy_static;
// use std::sync::Mutex;

// lazy_static! {
//     static ref VALUES: Mutex<Vec<i32>> = Mutex::default();
// }

// let mut values = VALUES.lock().unwrap();
// values.extend_from_slice(&[1, 2, 3, 4]);
// assert_eq!(&*values, &[1, 2, 3, 4]);