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



Sample input:
2 2 3
1 1 3
2 1 4
1 2 5


Sample output:
8

Submission status: Time Limit Exceeded

Solution code:
## coding: UTF-8
R,C,K = map(int,input().split())
rcv = []
for i in range(K):
    rcv.append(list(map(int,input().split())))

#print(rcv)

available = [False] * (R*C+1) #アイテムの有無を示す
item = [0] * (R*C+1)
for i in range(K):
    index = (rcv[i][0] - 1) * C + (rcv[i][1] - 1) +1
    available[index] = True
    item[index] = rcv[i][2]
#print(available, item)


def solver(R,C,K,rcv):
    #その場所にいる段階での最大スコアを出力
    #その行で獲得した個数も格納
    #DPの都合上0番目の要素を追加
    #グリッドは1次元で処理(1行目は1~C, 2行目はC+1~2C, ...)
    dp = []
    for i in range(R*C+1):
        dp.append([0] * 4)
    #print(dp)
    for i in range(1, R*C+1):
        #print(item[i])
        if(i == 1):
            #左上なら
            #dp[i][0] = dp[i-1][0] #左から(何も持ってこない)
            if(available[i]):
                #itemがある場合は取得するときと取得しない二通りに分けられる
                dp[i][1] = dp[i-1][0] + item[i]
        elif(i <= C):
            #左上以外の上(左からくる場合に限られる)
            dp[i][0] = dp[i-1][0] #itemを取得しない場合
            if(available[i]):
                dp[i][1] = max(dp[i-1][1], dp[i-1][0] + item[i])
                #if(dp[i-1][1] > 0): #左ですでに一個以上所持している場合にのみ、このマスで二個目ということが可能
                dp[i][2] = max(dp[i-1][2], dp[i-1][1] + item[i])
                #else:
                    #dp[i][2] = 0
                #if(dp[i-1][2] > 0): #左ですでに二個以上所持している場合にのみ、このマスで三個目ということが可能
                dp[i][3] = max(dp[i-1][3], dp[i-1][2] + item[i])
                #else:
                    #dp[i][3] = 0
            else:
                dp[i][1] = dp[i-1][1]
                dp[i][2] = dp[i-1][2]
                dp[i][3] = dp[i-1][3]
        elif(i % C == 1):
            #左上以外の左(上からくる場合に限られる)
            dp[i][0] = max(dp[i-C]) #下に降りてくる場合は、行で取得したアイテム数はリセットされる
            if(available[i]):
                dp[i][1] = max(dp[i-C]) + item[i]
        else:
            #左でも上でもない場合
            dp[i][0] = max(dp[i-1][0], max(dp[i-C])) #左からくる場合は0限定だが、上からくる場合は、上の行で何個拾っててもいい
            if(available[i]):
                dp[i][1] = max(dp[i-1][1], dp[i-1][0] + item[i], max(dp[i-C]) + item[i] ) #左から一個持った状態で来て拾わないor左から何も持たずに来て拾うor上から来て(所持数関係無し)拾う
                #if(dp[i-1][1] > 0): #左ですでに一個以上所持している場合にのみ、このマスで二個目ということが可能
                dp[i][2] = max(dp[i-1][2], dp[i-1][1] + item[i]) #上から来て二個目は不可能
                #else:
                    #dp[i][2] = 0
                #if(dp[i-1][2] > 0):
                dp[i][3] = max(dp[i-1][3], dp[i-1][2] + item[i])
                #else:
                    #dp[i][3] = 0
            else:
                dp[i][1] = dp[i-1][1]
                dp[i][2] = dp[i-1][2]
                dp[i][3] = dp[i-1][3]

         

        #print(i, dp)
    print(max(dp[R*C]))


solver(R,C,K,rcv)

Submission status: Time Limit Exceeded

Solution code:
r,c,k=map(int,input().split())

vmap=[[0 for i in range(c)] for j in range(r)]

for i in range(k):
    ri,ci,vi=map(int,input().split())
    vmap[ri-1][ci-1]=vi

f=[[[0,0,0,0] for i in range(c)] for j in range(r)]

f[0][0][1]=vmap[0][0]
for i in range(r-1):
    f[i+1][0][0]=max(f[i][0])
    f[i+1][0][1]=max(f[i][0])+vmap[i+1][0]
    for j in range(1,c):
        f[i][j][0]=max(f[i][j][0],f[i][j-1][0])
        f[i][j][1]=max(f[i][j][1],f[i][j-1][1],f[i][j-1][0]+vmap[i][j])
        f[i][j][2]=max(f[i][j][2],f[i][j-1][2],f[i][j-1][1]+vmap[i][j])
        f[i][j][3]=max(f[i][j][3],f[i][j-1][3],f[i][j-1][2]+vmap[i][j])
        f[i+1][j][0]=max(f[i][j])
        f[i+1][j][1]=max(f[i][j])+vmap[i+1][j]

for j in range(1,c):
    f[r-1][j][0]=max(f[r-1][j][0],f[r-1][j-1][0])
    f[r-1][j][1]=max(f[r-1][j][1],f[r-1][j-1][1],f[r-1][j-1][0]+vmap[r-1][j])
    f[r-1][j][2]=max(f[r-1][j][2],f[r-1][j-1][2],f[r-1][j-1][1]+vmap[r-1][j])
    f[r-1][j][3]=max(f[r-1][j][3],f[r-1][j-1][3],f[r-1][j-1][2]+vmap[r-1][j])

print(max(f[r-1][c-1]))

Submission status: Time Limit Exceeded

Solution code:
## coding: UTF-8
R,C,K = map(int,input().split())
rcv = []
for i in range(K):
    rcv.append(list(map(int,input().split())))

#print(rcv)

available = [False] * (R*C+1) #アイテムの有無を示す
item = [0] * (R*C+1)
for i in range(K):
    index = (rcv[i][0] - 1) * C + (rcv[i][1] - 1) +1
    available[index] = True
    item[index] = rcv[i][2]
#print(available, item)


def solver(R,C,K,rcv):
    #その場所にいる段階での最大スコアを出力
    #その行で獲得した個数も格納
    #DPの都合上0番目の要素を追加
    #グリッドは1次元で処理(1行目は1~C, 2行目はC+1~2C, ...)
    dp = []
    for i in range(R*C+1):
        dp.append([0] * 4)
    #print(dp)
    for i in range(1, R*C+1):
        #print(item[i])
        if(i == 1):
            #左上なら
            #dp[i][0] = dp[i-1][0] #左から(何も持ってこない)
            if(available[i]):
                #itemがある場合は取得するときと取得しない二通りに分けられる
                dp[i][1] = dp[i-1][0] + item[i]
        elif(i <= C):
            #左上以外の上(左からくる場合に限られる)
            dp[i][0] = dp[i-1][0] #itemを取得しない場合
            if(available[i]):
                dp[i][1] = max(dp[i-1][1], dp[i-1][0] + item[i])
                if(dp[i-1][1] > 0): #左ですでに一個以上所持している場合にのみ、このマスで二個目ということが可能
                    dp[i][2] = max(dp[i-1][2], dp[i-1][1] + item[i])
                else:
                    dp[i][2] = 0
                if(dp[i-1][2] > 0): #左ですでに二個以上所持している場合にのみ、このマスで三個目ということが可能
                    dp[i][3] = max(dp[i-1][3], dp[i-1][2] + item[i])
                else:
                    dp[i][3] = 0
            else:
                dp[i][1] = dp[i-1][1]
                dp[i][2] = dp[i-1][2]
                dp[i][3] = dp[i-1][3]
        elif(i % C == 1):
            #左上以外の左(上からくる場合に限られる)
            dp[i][0] = max(dp[i-C]) #下に降りてくる場合は、行で取得したアイテム数はリセットされる
            if(available[i]):
                dp[i][1] = max(dp[i-C]) + item[i]
        else:
            #左でも上でもない場合
            dp[i][0] = max(dp[i-1][0], max(dp[i-C])) #左からくる場合は0限定だが、上からくる場合は、上の行で何個拾っててもいい
            if(available[i]):
                dp[i][1] = max(dp[i-1][1], dp[i-1][0] + item[i], max(dp[i-C]) + item[i] ) #左から一個持った状態で来て拾わないor左から何も持たずに来て拾うor上から来て(所持数関係無し)拾う
                if(dp[i-1][1] > 0): #左ですでに一個以上所持している場合にのみ、このマスで二個目ということが可能
                    dp[i][2] = max(dp[i-1][2], dp[i-1][1] + item[i]) #上から来て二個目は不可能
                else:
                    dp[i][2] = 0
                if(dp[i-1][2] > 0):
                    dp[i][3] = max(dp[i-1][3], dp[i-1][2] + item[i])
                else:
                    dp[i][3] = 0
            else:
                dp[i][1] = dp[i-1][1]
                dp[i][2] = dp[i-1][2]
                dp[i][3] = dp[i-1][3]

         

        #print(i, dp)
    print(max(dp[R*C]))


solver(R,C,K,rcv)

Submission status: Time Limit Exceeded

Solution code:
r,c,k_trash = map(int,input().split())


v = [[0 for _ in range(r)] for _ in range(c)]
dp = [[[0 for _ in range(r)] for _ in range(c)] for _ in range(4)]

for _ in range(k_trash):
    x,y,v_tmp = map(int,input().split())
    v[y-1][x-1] = v_tmp


dp[1][0][0] = v[0][0]
for i in range(r):
    for j in range(c):
        for k in range(4):

            if(i >= 1):
                dp[0][j][i] = max(dp[0][j][i],dp[k][j][i-1])
                dp[1][j][i] = max(dp[1][j][i],dp[k][j][i-1] + v[j][i])

            if(j >= 1):
                dp[k][j][i] = max(dp[k][j][i],dp[k][j-1][i])
                if(k >= 1):
                    dp[k][j][i] = max(dp[k][j][i],dp[k-1][j-1][i] + v[j][i])

ans = 0

for k in range(4):
    ans = max(ans,dp[k][c-1][r-1])

print(ans)

Submission status: Time Limit Exceeded

Solution code:
import sys
def input(): return sys.stdin.readline().rstrip()
def ii(): return int(input())
def mi(): return map(int, input().split())
def li(): return list(mi())

import numpy as np
def main():
    r, c, k = mi()
    g = np.zeros((r+1, c+1), dtype=int)
    for _ in range(k):
        i, j, v = mi()
        g[i][j] = v
    dp = np.zeros((r+1,c+1,4), dtype=int)
    for i in range(1, r+1):
        for j in range(1, c+1):
            for k in range(4):
                dp[i][j][k] = max(dp[i][j-1][k], dp[i-1][j][3])
            if g[i][j] > 0:
                for k in range(2, -1, -1):
                    dp[i][j][k+1] = max(dp[i][j][k+1], dp[i][j][k]+g[i][j])

    print(dp[r][c][3])


if __name__ == '__main__':
    main()

Submission status: Time Limit Exceeded

Solution code:
import copy
R, C, K = map(int,input().split())
i = [list(map(int,input().split())) for k in range(K)]
dp0 = [[0] * C for r in range(R)]
location = copy.deepcopy(dp0)
dp1 = copy.deepcopy(dp0)
dp2 = copy.deepcopy(dp0)
dp3 = copy.deepcopy(dp0)
for k in range(K):
   location[i[k][0]-1][i[k][1]-1] = i[k][2]
dp1[0][0] = location[0][0]
if R > 1:
   for r in range(R-1):
      dp1[r+1][0] = location[r+1][0] + dp1[r][0]
      dp0[r+1][0] = dp1[r][0]
if C > 1:
   for c in range(C-1):
      dp0[0][c+1] = dp0[0][c]
      dp1[0][c+1] = max(location[0][c+1] + dp0[0][c], dp1[0][c])
      dp2[0][c+1] = max(location[0][c+1] + dp1[0][c], dp2[0][c])
      dp3[0][c+1] = max(location[0][c+1] + dp2[0][c], dp3[0][c])
   if R > 1:
      for r in range(R-1):
         for c in range(C-1):
            if location[r+1][c+1] > 0:
               dp0[r+1][c+1] = max(dp0[r+1][c], dp0[r][c+1], dp1[r][c+1], dp2[r][c+1], dp3[r][c+1])
               dp1[r+1][c+1] = max(dp0[r+1][c+1] + location[r+1][c+1], dp1[r+1][c])
               dp2[r+1][c+1] = max(dp1[r+1][c] + location[r+1][c+1], dp2[r+1][c])
               dp3[r+1][c+1] = max(dp2[r+1][c] + location[r+1][c+1], dp3[r+1][c])
            else:
               dp0[r+1][c+1] = max(dp0[r+1][c], dp0[r][c+1], dp1[r][c+1], dp2[r][c+1], dp3[r][c+1])
               dp1[r+1][c+1] = max(dp0[r+1][c+1], dp1[r+1][c])
               dp2[r+1][c+1] = max(dp1[r+1][c], dp2[r+1][c])
               dp3[r+1][c+1] = max(dp2[r+1][c], dp3[r+1][c])
if R==1 and C==1:
   ans = i[-1][-1]
else:
   ans = max(dp0[R-1][C-1], dp1[R-1][C-1], dp2[R-1][C-1], dp3[R-1][C-1])
print(ans)

Submission status: Time Limit Exceeded

Solution code:
R, C, K = map(int, input().split())
B = [[0] * (C + 1) for _ in range(R + 1)]

for i in range(K):
    r, c, v = map(int, input().split())
    B[r][c] = v

dp = [[[0] * 4 for _ in range(C + 1)] for _ in range(R + 1)]

for r in range(1, R + 1):
    for c in range(1, C + 1):
        dp[r][c][0] = max(dp[r - 1][c])
        dp[r][c][0] = max(dp[r][c][0], dp[r][c - 1][0])
        dp[r][c][1] = max(dp[r - 1][c]) + B[r][c]
        dp[r][c][1] = max(dp[r][c][1], dp[r][c - 1][1], dp[r][c - 1][0] + B[r][c])
        dp[r][c][2] = max(dp[r][c - 1][2], dp[r][c - 1][1] + B[r][c])
        dp[r][c][3] = max(dp[r][c - 1][3], dp[r][c - 1][2] + B[r][c])

print(max(dp[R][C]))

Submission status: Time Limit Exceeded

Solution code:
import copy
R, C, K = map(int,input().split())
i = [list(map(int,input().split())) for k in range(K)]
dp0 = []
for r in range(R):
   dp0.append([])
   for c in range(C):
      dp0[r].append([])
      dp0[r][c] = 0
location = copy.deepcopy(dp0)
dp1 = copy.deepcopy(dp0)
dp2 = copy.deepcopy(dp0)
dp3 = copy.deepcopy(dp0)
for k in range(K):
   location[i[k][0]-1][i[k][1]-1] = i[k][2]
dp1[0][0] = location[0][0]
if R > 1:
   for r in range(R-1):
      dp1[r+1][0] = location[r+1][0] + dp1[r][0]
      dp0[r+1][0] = dp1[r][0]
if C > 1:
   for c in range(C-1):
      dp0[0][c+1] = dp0[0][c]
      dp1[0][c+1] = max(location[0][c+1] + dp0[0][c], dp1[0][c])
      dp2[0][c+1] = max(location[0][c+1] + dp1[0][c], dp2[0][c])
      dp3[0][c+1] = max(location[0][c+1] + dp2[0][c], dp3[0][c])
   if R > 1:
      for r in range(R-1):
         for c in range(C-1):
            if location[r+1][c+1] > 0:
               dp0[r+1][c+1] = max(dp0[r+1][c], dp0[r][c+1], dp1[r][c+1], dp2[r][c+1], dp3[r][c+1])
               dp1[r+1][c+1] = max(dp0[r+1][c+1] + location[r+1][c+1], dp1[r+1][c])
               dp2[r+1][c+1] = max(dp1[r+1][c] + location[r+1][c+1], dp2[r+1][c])
               dp3[r+1][c+1] = max(dp2[r+1][c] + location[r+1][c+1], dp3[r+1][c])
            else:
               dp0[r+1][c+1] = max(dp0[r+1][c], dp0[r][c+1], dp1[r][c+1], dp2[r][c+1], dp3[r][c+1])
               dp1[r+1][c+1] = max(dp0[r+1][c+1], dp1[r+1][c])
               dp2[r+1][c+1] = max(dp1[r+1][c], dp2[r+1][c])
               dp3[r+1][c+1] = max(dp2[r+1][c], dp3[r+1][c])
if R==1 and C==1:
   ans = i[-1][-1]
else:
   ans = max(dp0[R-1][C-1], dp1[R-1][C-1], dp2[R-1][C-1], dp3[R-1][C-1])
print(ans)

Submission status: Time Limit Exceeded

Solution code:
import copy
R, C, K = map(int,input().split())
i = [list(map(int,input().split())) for k in range(K)]
dp0 = []
for r in range(R):
   dp0.append([])
   for c in range(C):
      dp0[r].append([])
      dp0[r][c] = 0
location = copy.deepcopy(dp0)
dp1 = copy.deepcopy(dp0)
dp2 = copy.deepcopy(dp0)
dp3 = copy.deepcopy(dp0)
for k in range(K):
   location[i[k][0]-1][i[k][1]-1] = i[k][2]
dp1[0][0] = location[0][0]
if R > 1:
   for r in range(R-1):
      dp1[r+1][0] = location[r+1][0] + dp1[r][0]
      dp0[r+1][0] = dp1[r][0]
if C > 1:
   for c in range(C-1):
      dp0[0][c+1] = dp0[0][c]
      dp1[0][c+1] = max(location[0][c+1] + dp0[0][c], dp1[0][c])
      dp2[0][c+1] = max(location[0][c+1] + dp1[0][c], dp2[0][c])
      dp3[0][c+1] = max(location[0][c+1] + dp2[0][c], dp3[0][c])
   if R > 1:
      for r in range(R-1):
         for c in range(C-1):
            dp0[r+1][c+1] = max(dp0[r+1][c], dp0[r][c+1], dp1[r][c+1], dp2[r][c+1], dp3[r][c+1])
            dp1[r+1][c+1] = max(dp0[r+1][c+1] + location[r+1][c+1], dp1[r+1][c])
            dp2[r+1][c+1] = max(dp1[r+1][c] + location[r+1][c+1], dp2[r+1][c])
            dp3[r+1][c+1] = max(dp2[r+1][c] + location[r+1][c+1], dp3[r+1][c])
if R==1 and C==1:
   ans = i[-1][-1]
else:
   ans = max(dp0[R-1][C-1], dp1[R-1][C-1], dp2[R-1][C-1], dp3[R-1][C-1])
print(ans)

Submission status: Time Limit Exceeded

Solution code:
#!/usr/bin python3
# -*- coding: utf-8 -*-

r, c, k = map(int, input().split())
itm = [[0]*(c) for _ in range(r)]
for i in range(k):
    ri, ci, vi = map(int, input().split())
    itm[ri-1][ci-1] = vi

dp0 = [[0]*4 for c_ in range(3005)]
dp1 = [[0]*4 for c_ in range(3005)]

for i in range(r):
    for j in range(c):
        nowv = itm[i][j]
        dp0[j][3] = max(dp0[j][3], dp0[j][2] + nowv)
        dp0[j][2] = max(dp0[j][2], dp0[j][1] + nowv)
        dp0[j][1] = max(dp0[j][1], dp0[j][0] + nowv)

        dp0[j+1][0] = max(dp0[j+1][0], dp0[j][0])
        dp0[j+1][1] = max(dp0[j+1][1], dp0[j][1])
        dp0[j+1][2] = max(dp0[j+1][2], dp0[j][2])
        dp0[j+1][3] = max(dp0[j+1][3], dp0[j][3])

        dp1[j][0] = max(dp1[j][0], max(dp0[j]))
    dp0 = dp1.copy()

print(max(dp1[c-1]))

Submission status: Time Limit Exceeded

Solution code:
#!/usr/bin python3
# -*- coding: utf-8 -*-

import numpy as np
from numba import jit



r, c, k = map(int, input().split())
itm = [[0]*(c) for _ in range(r)]
for i in range(k):
    ri, ci, vi = map(int, input().split())
    itm[ri-1][ci-1] = vi


@jit
def f(r,c,itm):
    dp = np.zeros((3005, 3005, 4), dtype=np.int64)
    for i in range(r):
        for j in range(c):
            nowv = itm[i][j]
            dp[i][j][1:] = np.maximum(dp[i][j][1:], dp[i][j][:3] + nowv)
            dp[i+1][j][0] = max(dp[i+1][j][0], max(dp[i][j]))
            dp[i][j+1][:] = np.maximum(dp[i][j+1][:], dp[i][j][:])
    print(max(dp[r-1][c-1]))

f(r,c,itm)

Submission status: Time Limit Exceeded

Solution code:
#!/usr/bin python3
# -*- coding: utf-8 -*-

import numpy as np
from numba import jit



r, c, k = map(int, input().split())
itm = [[0]*(c) for _ in range(r)]
for i in range(k):
    ri, ci, vi = map(int, input().split())
    itm[ri-1][ci-1] = vi

dp = np.zeros((3005, 3005, 4), dtype=np.int64)

@jit
def f(i, j, x, dp):
    dp[i][j][1:] = np.maximum(dp[i][j][1:], dp[i][j][:3] + x)
    dp[i+1][j][0] = max(dp[i+1][j][0], max(dp[i][j]))
    dp[i][j+1][:] = np.maximum(dp[i][j+1][:], dp[i][j][:])


for i in range(r):
    for j in range(c):
        nowv = itm[i][j]
        f(i, j, nowv, dp)
#        dp[i][j][1:] = np.maximum(dp[i][j][1:], dp[i][j][:3] + nowv)
#        dp[i+1][j][0] = max(dp[i+1][j][0], max(dp[i][j]))
#        dp[i][j+1][:] = np.maximum(dp[i][j+1][:], dp[i][j][:])

print(max(dp[r-1][c-1]))

Submission status: Time Limit Exceeded

Solution code:
#!/usr/bin python3
# -*- coding: utf-8 -*-

import numpy as np

r, c, k = map(int, input().split())
itm = [[0]*(c) for _ in range(r)]
for i in range(k):
    ri, ci, vi = map(int, input().split())
    itm[ri-1][ci-1] = vi

dp = np.zeros((3005, 3005, 4), dtype=np.int64)

for i in range(r):
    for j in range(c):
        nowv = itm[i][j]
        dp[i][j][1:] = np.maximum(dp[i][j][1:], dp[i][j][:3] + nowv)
#        dp[i][j][3] = max(dp[i][j][3], dp[i][j][2] + nowv)
#        dp[i][j][2] = max(dp[i][j][2], dp[i][j][1] + nowv)
#        dp[i][j][1] = max(dp[i][j][1], dp[i][j][0] + nowv)
#
        dp[i+1][j][0] = max(dp[i+1][j][0], max(dp[i][j]))
        dp[i][j+1][:] = np.maximum(dp[i][j+1][:], dp[i][j][:])

#        dp[i][j+1][0] = max(dp[i][j+1][0], dp[i][j][0])
#        dp[i][j+1][1] = max(dp[i][j+1][1], dp[i][j][1])
#        dp[i][j+1][2] = max(dp[i][j+1][2], dp[i][j][2])
#        dp[i][j+1][3] = max(dp[i][j+1][2], dp[i][j][3])

print(max(dp[r-1][c-1]))

Submission status: Time Limit Exceeded

Solution code:
#!/usr/bin python3
# -*- coding: utf-8 -*-

r, c, k = map(int, input().split())
itm = [[0]*(c) for _ in range(r)]
for i in range(k):
    ri, ci, vi = map(int, input().split())
    itm[ri-1][ci-1] = vi

dp = [[[0]*4 for i in range(3005)] for j in range(3005)]

for i in range(r):
    for j in range(c):
        nowv = itm[i][j]
        dp[i][j][3] = max(dp[i][j][3], dp[i][j][2] + nowv)
        dp[i][j][2] = max(dp[i][j][2], dp[i][j][1] + nowv)
        dp[i][j][1] = max(dp[i][j][1], dp[i][j][0] + nowv)

        dp[i+1][j][0] = max(dp[i+1][j][0], max(dp[i][j]))
        dp[i][j+1][0] = max(dp[i][j+1][0], dp[i][j][0])
        dp[i][j+1][1] = max(dp[i][j+1][1], dp[i][j][1])
        dp[i][j+1][2] = max(dp[i][j+1][2], dp[i][j][2])
        dp[i][j+1][3] = max(dp[i][j+1][2], dp[i][j][3])

print(max(dp[r-1][c-1]))

Submission status: Time Limit Exceeded

Solution code:
#!/usr/bin python3
# -*- coding: utf-8 -*-

import numpy as np

r, c, k = map(int, input().split())
itm = [[0]*(c) for _ in range(r)]
for i in range(k):
    ri, ci, vi = map(int, input().split())
    itm[ri-1][ci-1] = vi

dp = np.zeros((3005, 3005, 4), dtype=np.int64)

for i in range(r):
    for j in range(c):
        nowv = itm[i][j]
        dp[i][j][3] = max(dp[i][j][3], dp[i][j][2] + nowv)
        dp[i][j][2] = max(dp[i][j][2], dp[i][j][1] + nowv)
        dp[i][j][1] = max(dp[i][j][1], dp[i][j][0] + nowv)

        dp[i+1][j][0] = max(dp[i+1][j][0], max(dp[i][j]))
        dp[i][j+1][0] = max(dp[i][j+1][0], dp[i][j][0])
        dp[i][j+1][1] = max(dp[i][j+1][1], dp[i][j][1])
        dp[i][j+1][2] = max(dp[i][j+1][2], dp[i][j][2])
        dp[i][j+1][3] = max(dp[i][j+1][2], dp[i][j][3])

print(max(dp[r-1][c-1]))

Submission status: Time Limit Exceeded

Solution code:
R, C, K = map(int, input().split())
item = [[0 for _ in range(C + 1)] for _ in range(R + 1)]
for _ in range(K):
    r, c, v = map(int, input().split())
    item[r][c] = v

# 使いたい引数を「逆順」で書くと上手く多次元DP配列が生成できる
dp = [[[0 for _ in range(4)] for _ in range(C + 1)] for _ in range(R + 1)]

for i in range(R + 1):
    for j in range(C + 1):
        for k in range(4):
            now = dp[i][j][k]
            # 下に移動する場合、アイテムの個数はリセットされる
            # i == Rの時は下に移動できない
            if i <= R - 1:
                # 移動先のアイテムを取らない場合
                dp[i + 1][j][0] = max(dp[i + 1][j][0], now)
                # 移動先のアイテムを取る場合
                dp[i + 1][j][1] = max(dp[i + 1][j][1], now + item[i + 1][j])
            # 右に移動する場合、アイテムの個数は維持される
            # j == Cの時は右に移動できない
            if j <= C - 1:
                dp[i][j + 1][k] = max(dp[i][j + 1][k], now)
                # 現在のkが3未満(0 or 1 or 2)の場合、移動先のアイテムをとることが可能
                if k < 3:
                    dp[i][j + 1][k + 1] = max(dp[i][j + 1]
                                              [k + 1], now + item[i][j + 1])

ans = 0
# 最終的にR行で取るアイテムの個数についてそれぞれ調べる、3個が最適とは限らない
for k in range(4):
    ans = max(ans, dp[R][C][k])

print(ans)

Submission status: Time Limit Exceeded

Solution code:
import sys
input=sys.stdin.readline

class list2D:
    def __init__(self, H, W, num):
        self.__H = H
        self.__W = W
        self.__dat = [num] * (H * W)

    def __getitem__(self, a):
        return self.__dat[a[0]*self.__W+a[1]]

    def __setitem__(self, a, b):
        self.__dat[a[0]*self.__W+a[1]] = b

    def debug(self):
        print(self.__dat)

class list3D:
    def __init__(self, H, W, D, num):
        self.__H = H
        self.__W = W
        self.__D = D
        self.__dat = [num] * (H * W * D)

    def __getitem__(self, a):
        return self.__dat[a[0]*self.__W*self.__D+a[1]*self.__D + a[2]]

    def __setitem__(self, a, b):
        self.__dat[a[0]*self.__W*self.__D+a[1]*self.__D + a[2]] = b

    def debug(self):
        print(self.__dat)

def main():
    r,c,k=map(int,input().split())
    v = list2D(r, c, 0)

    for _ in range(k):
        ri,ci,a=map(int,input().split())
        v[ri-1, ci-1] = a

    dp = list3D(r, c, 4, 0)
    #print(dp)
    if v[0, 0]>0: dp[0, 0, 1]=v[0, 0]

    for i in range(r):
        for j in range(c):
            val = v[i, j]
            if i>0:
                x = max(dp[i-1, j, 0], dp[i-1, j, 1], dp[i-1, j, 2], dp[i-1, j, 3]) + val
                dp[i, j, 1]=max(dp[i, j, 1], x)
                dp[i, j, 0]=max(dp[i, j, 0],dp[i-1, j, 0],dp[i-1, j, 1],dp[i-1, j, 2],dp[i-1, j, 3])
            if j>0:
                dp[i, j, 0]=max(dp[i, j, 0],dp[i, j-1, 0])
                dp[i, j, 1]=max(dp[i, j, 1],dp[i, j-1, 0]+val,dp[i, j-1, 1])
                dp[i, j, 2]=max(dp[i, j, 2],dp[i, j-1, 1]+val,dp[i, j-1, 2])
                dp[i, j, 3]=max(dp[i, j, 3],dp[i, j-1, 2]+val,dp[i, j-1, 3])
    #print(dp)
    ans=0
    for i in range(4):
        ans=max(dp[r-1, c-1, i],ans)
    return print(ans)
if __name__=="__main__":
    main()

Submission status: Time Limit Exceeded

Solution code:
import sys
input=sys.stdin.readline

class list2D:
    def __init__(self, H, W, num):
        self.__H = H
        self.__W = W
        self.__dat = [num] * (H * W)

    def __getitem__(self, a):
        return self.__dat[a[0]*self.__W+a[1]]

    def __setitem__(self, a, b):
        self.__dat[a[0]*self.__W+a[1]] = b

    def debug(self):
        print(self.__dat)

class list3D:
    def __init__(self, H, W, D, num):
        self.__H = H
        self.__W = W
        self.__D = D
        self.__dat = [num] * (H * W * D)

    def __getitem__(self, a):
        return self.__dat[a[0]*self.__W*self.__D+a[1]*self.__D + a[2]]

    def __setitem__(self, a, b):
        self.__dat[a[0]*self.__W*self.__D+a[1]*self.__D + a[2]] = b

    def debug(self):
        print(self.__dat)

def main():
    r,c,k=map(int,input().split())
    v = list2D(r, c, 0)

    for _ in range(k):
        ri,ci,a=map(int,input().split())
        v[ri-1, ci-1] = a

    dp = list3D(r, c, 4, 0)
    #print(dp)
    if v[0, 0]>0: dp[0, 0, 1]=v[0, 0]

    for i in range(r):
        for j in range(c):
            if i>0:
                dp[i, j, 1]=max(dp[i, j, 1],dp[i-1, j, 0]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i-1, j, 0])
                dp[i, j, 1]=max(dp[i, j, 1],dp[i-1, j, 1]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i-1, j, 1])
                dp[i, j, 1]=max(dp[i, j, 1],dp[i-1, j, 2]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i-1, j, 2])
                dp[i, j, 1]=max(dp[i, j, 1],dp[i-1, j, 3]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i-1, j, 3])
            if j>0:
                dp[i, j, 1]=max(dp[i, j, 1],dp[i, j-1, 0]+v[i, j])
                dp[i, j, 2]=max(dp[i, j, 2],dp[i, j-1, 1]+v[i, j])
                dp[i, j, 3]=max(dp[i, j, 3],dp[i, j-1, 2]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i, j-1, 0])
                dp[i, j, 1]=max(dp[i, j, 1],dp[i, j-1, 1])
                dp[i, j, 2]=max(dp[i, j, 2],dp[i, j-1, 2])
                dp[i, j, 3]=max(dp[i, j, 3],dp[i, j-1, 3])
    #print(dp)
    ans=0
    for i in range(4):
        ans=max(dp[r-1, c-1, i],ans)
    return print(ans)
if __name__=="__main__":
    main()

Submission status: Time Limit Exceeded

Solution code:
import sys
input=sys.stdin.readline

def main():
    r,c,k=map(int,input().split())
    v=[0] * (c * r)
    for _ in range(k):
        ri,ci,a=map(int,input().split())
        v[(ri-1)*c + (ci-1)]=a

    dp= [0] * (c * r * 4)

    #print(dp)
    x = c * r
    if v[0]>0: dp[x] = v[0]

    for i in range(r):
        for j in range(c):
            idx = i*c+j
            idx2 = (i-1)*c + j
            idx3 = i*c + j-1

            if i>0:
                a = max(dp[idx2],dp[x + idx2],dp[2*x+idx2],dp[3*x+idx2])+v[idx]
                dp[x+idx]=max(dp[x+idx], a)
                dp[idx]=max(dp[idx],dp[idx2],dp[x+idx2],dp[2*x+idx2],dp[3*x+idx2])

            if j>0:
                dp[idx]=max(dp[idx],dp[idx3])
                dp[x + idx]=max(dp[x + idx],dp[idx3]+v[idx],dp[x + idx3])
                dp[2*x + idx]=max(dp[2*x + idx],dp[x + idx3]+v[idx],dp[2*x + idx3])
                dp[3*x + idx]=max(dp[3*x + idx],dp[2*x + idx3]+v[idx],dp[3*x + idx3])

    #print(dp)
    ans=0
    for i in range(4): ans = max(dp[i*x + (r-1)*c + (c-1)],ans)
    return print(ans)

if __name__=="__main__":
    main()

Submission status: Time Limit Exceeded

Solution code:
import sys
input=sys.stdin.readline

class list2D:
    def __init__(self, H, W, num):
        self.__H = H
        self.__W = W
        self.__dat = [num] * (H * W)

    def __getitem__(self, a):
        if a[0] < 0 or a[1] < 0 or a[0] >= self.__H or a[1] >= self.__W:
            print("Out of Index")
            exit(1)
        return self.__dat[a[0]*self.__W+a[1]]

    def __setitem__(self, a, b):
        if a[0] < 0 or a[1] < 0 or a[0] >= self.__H or a[1] >= self.__W:
            print("Out of Index")
            exit(1)
        self.__dat[a[0]*self.__W+a[1]] = b

    def debug(self):
        print(self.__dat)

class list3D:
    def __init__(self, H, W, D, num):
        self.__H = H
        self.__W = W
        self.__D = D
        self.__dat = [num] * (H * W * D)

    def __getitem__(self, a):
        if a[0] < 0 or a[1] < 0 or a[0] >= self.__H or a[1] >= self.__W or a[2] < 0 or a[2] >= self.__D:
            print("Out of Index")
            exit(1)
        return self.__dat[a[0]*self.__W*self.__D+a[1]*self.__D + a[2]]

    def __setitem__(self, a, b):
        if a[0] < 0 or a[1] < 0 or a[0] >= self.__H or a[1] >= self.__W or a[2] < 0 or a[2] >= self.__D:
            print("Out of Index")
            exit(1)
        self.__dat[a[0]*self.__W*self.__D+a[1]*self.__D + a[2]] = b

    def debug(self):
        print(self.__dat)

def main():
    r,c,k=map(int,input().split())
    v = list2D(r, c, 0)

    for _ in range(k):
        ri,ci,a=map(int,input().split())
        v[ri-1, ci-1] = a

    dp = list3D(r, c, 4, 0)
    #print(dp)
    if v[0, 0]>0: dp[0, 0, 1]=v[0, 0]

    for i in range(r):
        for j in range(c):
            if i>0:
                dp[i, j, 1]=max(dp[i, j, 1],dp[i-1, j, 0]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i-1, j, 0])
                dp[i, j, 1]=max(dp[i, j, 1],dp[i-1, j, 1]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i-1, j, 1])
                dp[i, j, 1]=max(dp[i, j, 1],dp[i-1, j, 2]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i-1, j, 2])
                dp[i, j, 1]=max(dp[i, j, 1],dp[i-1, j, 3]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i-1, j, 3])
            if j>0:
                dp[i, j, 1]=max(dp[i, j, 1],dp[i, j-1, 0]+v[i, j])
                dp[i, j, 2]=max(dp[i, j, 2],dp[i, j-1, 1]+v[i, j])
                dp[i, j, 3]=max(dp[i, j, 3],dp[i, j-1, 2]+v[i, j])
                dp[i, j, 0]=max(dp[i, j, 0],dp[i, j-1, 0])
                dp[i, j, 1]=max(dp[i, j, 1],dp[i, j-1, 1])
                dp[i, j, 2]=max(dp[i, j, 2],dp[i, j-1, 2])
                dp[i, j, 3]=max(dp[i, j, 3],dp[i, j-1, 3])
    #print(dp)
    ans=0
    for i in range(4):
        ans=max(dp[r-1, c-1, i],ans)
    return print(ans)
if __name__=="__main__":
    main()

Submission status: Time Limit Exceeded

Solution code:
import sys
input = sys.stdin.readline

r, c, k = map(int, input().split())
v = [[0 for j in range(c)] for i in range(r)]

dp = [[[0 for k in range(4)] for j in range(c)] for i in range(r)]

for i in range(k):
    ri, ci, vi = map(int, input().split())
    v[ri-1][ci-1] = vi


for i in range(r):
    if i > 0:
        for j in range(c):
            for k in range(4):
                dp[i][j][0] = max(dp[i][j][0], dp[i-1][j][k])
    for j in range(c):
        if j > 0:
            for k in range(4):
                dp[i][j][k] = max(dp[i][j][k], dp[i][j-1][k])
        if v[i][j]:
            for k in reversed(range(1, 4)):
                dp[i][j][k] = max(dp[i][j][k], dp[i][j][k-1]+v[i][j])
        

ans = 0

for k in range(4):
    ans = max(ans, dp[r-1][c-1][k])

print(ans)

Submission status: Time Limit Exceeded

Solution code:
ma = lambda :map(int,input().split())
lma = lambda :list(map(int,input().split()))
tma = lambda :tuple(map(int,input().split()))
ni = lambda:int(input())
yn = lambda fl:print("Yes") if fl else print("No")
import collections
import math
import itertools
import heapq as hq
R,C,K = ma()
V = [[0 for i in range(R+1)] for j in range(C+1)]
for i in range(K):
    r,c,v = ma()
    V[c][r] = v
dp_prev = [[0]*4 for i in range(C+1)]
dp_nex = [[0]*4 for i in range(C+1)] #dp[h][k] h列目でmax kこまで選んだ時の最大値
for r in range(1,R+1):
    for c in range(1,C+1):
        v = V[c][r]
        for k in range(4):
            if k==0:
                dp_nex[c][0] = max(dp_prev[c][3],dp_nex[c-1][0])
            elif k >0:
                dp_nex[c][k] = max(dp_nex[c-1][k],dp_nex[c-1][k-1]+v,dp_prev[c][3]+v)
    dp_prev,dp_nex = dp_nex,dp_prev
    #print(dp_prev)


print(dp_prev[C][3])