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: Wrong Answer

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

dp = [[[0]*4 for _ in range(C)] for _ in range(R)]
value = [[0]*3 for _ in range(R)]
for i in range(R):
    for j in range(C):
        if i == 0 and j == 0 and L[i][j] > 0:
            dp[i][j][1] = L[i][j]
        elif i == 0:
            for k in range(4):
                dp[i][j][k] = dp[i][j-1][k]
            if L[i][j] > 0:
                for k in range(3):
                    dp[i][j][k+1] = max(dp[i][j][k+1], dp[i][j-1][k]+L[i][j])
        elif j == 0:
            dp[i][j][0] = max(dp[i-1][j])
        else:
            dp[i][j][0] = max(max(dp[i-1][j]), dp[i][j-1][0])
            for k in range(1, 4):
                dp[i][j][k] = dp[i][j-1][k]
            if L[i][j] > 0:
                for k in range(3):
                    dp[i][j][k+1] = max(dp[i][j][k+1], dp[i][j-1][k]+L[i][j])
                dp[i][j][1] = max(dp[i][j][1], dp[i][j][0]+L[i][j])

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

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
value=[ [0 for _ in range(C+1)] for __ in range(R+1) ]
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[r][c]=v

dp0=[float("-inf")]*(C+1)
dp1=[float("-inf")]*(C+1)
dp2=[float("-inf")]*(C+1)
dp3=[float("-inf")]*(C+1)

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[1][1]

for r in range(1,R+1):
    for c in range(2,C+1):
        dp0[c]=max( dp0[c-1], dp0[c] )
        dp1[c]=max( dp1[c-1], dp1[c] )
        dp2[c]=max( dp2[c-1], dp2[c] )
        dp3[c]=max( dp3[c-1], dp3[c] )
        #今のマスにアイテムがある場合
        if 0<value[r][c]:
            dp3[c]=max( dp3[c-1], dp2[c]+value[r][c] )
            dp2[c]=max( dp2[c-1], dp1[c]+value[r][c] )
            dp1[c]=max( dp1[c-1], dp0[c]+value[r][c] )
        
    #上の行にアイテムをコピー
    for c in range(1,C+1):
        dp0[c]=max( dp0[c], dp1[c], dp2[c], dp3[c] )
    dp1=[float("-inf")]*(C+1)
    dp2=[float("-inf")]*(C+1)
    dp3=[float("-inf")]*(C+1)

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
value=[ [0 for _ in range(C+1)] for __ in range(R+1) ]
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[r][c]=v

dp0=[float("-inf")]*(C+1)
dp1=[float("-inf")]*(C+1)
dp2=[float("-inf")]*(C+1)
dp3=[float("-inf")]*(C+1)

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[1][1]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[r][c]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            #dp1[c]=max( dp1[c-1], dp0[c]+value[r][c])
            dp1[c]=max( dp1[c-1], dp0[c-1]+value[r][c], dp0[c]+value[r][c] )
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1]+value[r][c] )
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1]+value[r][c] )
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]

    #上の行にアイテムをコピー
    for c in range(1,C+1):
        dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    dp1=[float("-inf")]*(C+1)
    dp2=[float("-inf")]*(C+1)
    dp3=[float("-inf")]*(C+1)

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
value=[ [0 for _ in range(C+1)] for __ in range(R+1) ]
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[r][c]=v

dp0=[float("-inf")]*(C+1)
dp1=[float("-inf")]*(C+1)
dp2=[float("-inf")]*(C+1)
dp3=[float("-inf")]*(C+1)

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[1][1]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[r][c]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c]+value[r][c])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1]+value[r][c])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1]+value[r][c])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    for c in range(1,C+1):
        dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    dp1=[float("-inf")]*(C+1)
    dp2=[float("-inf")]*(C+1)
    dp3=[float("-inf")]*(C+1)

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
value=[ [0 for _ in range(C+1)] for __ in range(R+1) ]
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[r][c]=v

dp0=[0]*(C+1)
dp1=[0]*(C+1)
dp2=[0]*(C+1)
dp3=[0]*(C+1)

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[1][1]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[r][c]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c]+value[r][c])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1]+value[r][c])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1]+value[r][c])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    for c in range(1,C+1):
        dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    dp1=[0]*(C+1)
    dp2=[0]*(C+1)
    dp3=[0]*(C+1)

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
value=[ [0 for _ in range(C+1)] for __ in range(R+1) ]
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[r][c]=v

dp0=[0]*(C+1)
dp1=[0]*(C+1)
dp2=[0]*(C+1)
dp3=[0]*(C+1)

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[1][1]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[r][c]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c]+value[r][c])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1]+value[r][c])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1]+value[r][c])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    for c in range(1,C+1):
        dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
value=[ [0 for _ in range(C+1)] for __ in range(R+1) ]
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[r][c]=v
print value

dp0=[0]*(C+1)
dp1=[0]*(C+1)
dp2=[0]*(C+1)
dp3=[0]*(C+1)

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[1][1]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[r][c]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    #for c in range(1,C+1):
    #    dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    #dp1=defaultdict(lambda: float("-inf")) 
    #dp2=defaultdict(lambda: float("-inf")) 
    #dp3=defaultdict(lambda: float("-inf")) 

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp0=[0]*(C+1)
dp1=[0]*(C+1)
dp2=[0]*(C+1)
dp3=[0]*(C+1)

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[(1,1)]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        #if 0<value[(r,c)]:
        if True:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    #for c in range(1,C+1):
    #    dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    #dp1=defaultdict(lambda: float("-inf")) 
    #dp2=defaultdict(lambda: float("-inf")) 
    #dp3=defaultdict(lambda: float("-inf")) 

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp0=[0]*(C+1)
dp1=[0]*(C+1)
dp2=[0]*(C+1)
dp3=[0]*(C+1)

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[(1,1)]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[(r,c)]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    #for c in range(1,C+1):
    #    dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    #dp1=defaultdict(lambda: float("-inf")) 
    #dp2=defaultdict(lambda: float("-inf")) 
    #dp3=defaultdict(lambda: float("-inf")) 

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp0=defaultdict(lambda: float("-inf")) 
dp1=defaultdict(lambda: float("-inf")) 
dp2=defaultdict(lambda: float("-inf")) 
dp3=defaultdict(lambda: float("-inf")) 

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[(1,1)]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[(r,c)]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    #for c in range(1,C+1):
    #    dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    #dp1=defaultdict(lambda: float("-inf")) 
    #dp2=defaultdict(lambda: float("-inf")) 
    #dp3=defaultdict(lambda: float("-inf")) 

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp0=defaultdict(lambda: float("-inf")) 
dp1=defaultdict(lambda: float("-inf")) 
dp2=defaultdict(lambda: float("-inf")) 
dp3=defaultdict(lambda: float("-inf")) 

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[(1,1)]

for r in range(1,R+1):
    for c in range(2,C+1):
        pass
print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp0=defaultdict(lambda: float("-inf")) 
dp1=defaultdict(lambda: float("-inf")) 
dp2=defaultdict(lambda: float("-inf")) 
dp3=defaultdict(lambda: float("-inf")) 

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[(1,1)]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[(r,c)]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c]+value[(r,c)])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1]+value[(r,c)])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1]+value[(r,c)])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    #for c in range(1,C+1):
    #    dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    #dp1=defaultdict(lambda: float("-inf")) 
    #dp2=defaultdict(lambda: float("-inf")) 
    #dp3=defaultdict(lambda: float("-inf")) 

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp0=defaultdict(lambda: float("-inf")) 
dp1=defaultdict(lambda: float("-inf")) 
dp2=defaultdict(lambda: float("-inf")) 
dp3=defaultdict(lambda: float("-inf")) 

#初期化
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[(1,1)]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[(r,c)]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c]+value[(r,c)])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1]+value[(r,c)])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1]+value[(r,c)])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    for c in range(1,C+1):
        dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    #dp1=defaultdict(lambda: float("-inf")) 
    #dp2=defaultdict(lambda: float("-inf")) 
    #dp3=defaultdict(lambda: float("-inf")) 

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp0=defaultdict(lambda: float("-inf")) 
dp1=defaultdict(lambda: float("-inf")) 
dp2=defaultdict(lambda: float("-inf")) 
dp3=defaultdict(lambda: float("-inf")) 

#初期化  dp[(左から何列目,その行で何個アイテムを拾った)]
for c in range(1,C+1):
    dp0[c]=0
dp1[1]=value[(1,1)]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[(r,c)]:
            #今のマスまでで1個取得=max(左のマスまでで1個取得, 今のマスで0個取得+v )
            dp1[c]=max( dp1[c-1], dp0[c]+value[(r,c)])
            #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
            dp2[c]=max( dp2[c-1], dp1[c-1]+value[(r,c)])
            #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
            dp3[c]=max( dp3[c-1], dp2[c-1]+value[(r,c)])
        #今のマスにアイテムがない場合
        else:
            dp1[c]=dp1[c-1]
            dp2[c]=dp2[c-1]
            dp3[c]=dp3[c-1]
        
    #上の行にアイテムをコピー
    for c in range(1,C+1):
        dp0[c]=max(dp0[c],dp1[c],dp2[c],dp3[c])
    dp1=defaultdict(lambda: float("-inf")) 
    dp2=defaultdict(lambda: float("-inf")) 
    dp3=defaultdict(lambda: float("-inf")) 

print dp0[C]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp=defaultdict(lambda: float("-inf")) 
#初期化  dp[(左から何列目,その行で何個アイテムを拾った)]
dp[(1,0)]=0
#1,1のマスにアイテムがあるなら
if 0<value[(1,1)]:
    dp[(1,1)]=value[(1,1)]

for r in range(1,R+1):
    for c in range(2,C+1):
        #今のマスまでで3個取得=max(左のマスまでで3個取得, 左のマスまでで2個取得+v)
        dp[(c,3)]=max( dp[(c-1,3)], dp[(c-1,2)]+value[(r,c)] )
        #今のマスまでで2個取得=max(左のマスまでで2個取得, 左のマスまでで1個取得+v)
        dp[(c,2)]=max( dp[(c-1,2)], dp[(c-1,1)]+value[(r,c)] )
        #今のマスまでで1個取得=max(左のマスまでで1個取得, 左のマスまでで0個取得+v, 下のマスからの繰り上がり+v )
        dp[(c,1)]=max( dp[(c-1,1)], dp[(c-1,0)]+value[(r,c)], dp[(c,0)]+value[(r,c)] )
        #今のマスまでで0個取得=max(左のマスまでで0個取得, 下のマスからの繰り上がり+v )
        dp[(c,0)]=max( dp[(c-1,0)], dp[(c,0)] )

    #dpの値を上の行にコピー
    new_dp=defaultdict(lambda: float("-inf")) 
    for c in range(1,C+1):
        max_v=float("-inf")
        for k in range(0,4):
            max_v=max(max_v,dp[(c,k)])
        new_dp[(c,0)]=max_v

    dp=new_dp

print dp[(C,0)]

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp=defaultdict(lambda: float("-inf")) 
#初期化
dp[(1,1,0)]=0
if 0<value[(1,1)]:
    dp[(1,1,1)]=value[(1,1)]


#1番下の行だけdpテーブルを埋める
r=1
for c in range(2,C+1):
    #今のマスにアイテムがある場合
    if 0<value[(r,c)]:
        #今のマスまでで0個取得=今のマスの左横までで0個取得
        dp[(r,c,0)]=dp[(r,c-1,0)]
        #今のマスまでで1個取得=max(今のマスの左横までで1個取得, 今のマスまでで0個取得+v)
        dp[(r,c,1)]=max( dp[(r,c-1,1)],dp[(r,c-1,0)]+value[(r,c)] )
        #今のマスまでで2個取得=max(今のマスの左横までで2個取得, 今のマスまでで1個取得+v)
        dp[(r,c,2)]=max( dp[(r,c-1,2)],dp[(r,c-1,1)]+value[(r,c)] )
        #今のマスまでで3個取得=max(今のマスの左横までで3個取得, 今のマスまでで2個取得+v)
        dp[(r,c,3)]=max( dp[(r,c-1,3)],dp[(r,c-1,2)]+value[(r,c)] )
    #今のマスにアイテムがない場合
    else:
        #左横のマスの値をスライド
        dp[(r,c,0)]=dp[(1,c-1,0)]
        dp[(r,c,1)]=dp[(r,c-1,1)]
        dp[(r,c,2)]=dp[(r,c-1,2)]
        dp[(r,c,3)]=dp[(r,c-1,3)]


#1番左の列のdpテーブルを埋める
c=1
for r in range(2,R+1):
    #今のマスにアイテムがある場合
    if 0<value[(r,c)]:
        #今のマスまでで0個取得=max(下のマスで0個取得、下のマスで1個取得)
        dp[(r,c,0)]=max(dp[(r-1,c,0)],dp[(r-1,c,1)])
        #今のマスまでで1個取得=max(下のマスで0個取得+v、下のマスで1個取得+v)
        dp[(r,c,1)]=max(dp[(r-1,c,0)]+value[(r,c)],dp[(r-1,c,1)]+value[(r,c)])
    #今のマスにアイテムがない場合
    else:
        #今のマスまでで0個取得=max(下のマスで0個取得、下のマスで1個取得)
        dp[(r,c,0)]=max(dp[(r-1,c,0)],dp[(r-1,c,1)])


for r in range(2,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[(r,c)]:
            #今のマスまでで0個取得=max(今のマスの左横までで0個取得, 下のマスまでで0個取得, 下のマスまでで1個取得, 下のマスまでで2個取得, 下のマスまでで3個取得)
            dp[(r,c,0)]=max( dp[(r,c-1,0)], dp[(r-1,c,0)], dp[(r-1,c,1)], dp[(r-1,c,2)], dp[(r-1,c,3)] )
            #今のマスまでで1個取得=max(今のマスの左横までで1個取得, 今のマスまでで0個取得+v, 下のマスまでで0個取得+v, 下のマスまでで1個取得+v, 下のマスまでで2個取得+v, 下のマスまでで3個取得+v)
            dp[(r,c,1)]=max( dp[(r,c-1,1)], dp[(r,c-1,0)]+value[(r,c)], dp[(r-1,c,0)]+value[(r,c)], dp[(r-1,c,1)]+value[(r,c)], dp[(r-1,c,2)]+value[(r,c)], dp[(r-1,c,3)]+value[(r,c)]  )
            #今のマスまでで2個取得=max(今のマスの左横までで2個取得, 今のマスまでで1個取得+v)
            dp[(r,c,2)]=max( dp[(r,c-1,2)], dp[(r,c-1,1)]+value[(r,c)] )
            #今のマスまでで3個取得=max(今のマスの左横までで3個取得, 今のマスまでで2個取得+v)
            dp[(r,c,3)]=max( dp[(r,c-1,3)],dp[(r,c-1,2)]+value[(r,c)] )
        #今のマスにアイテムがない場合
        else:
            #今のマスまでで0個取得=max(今のマスの左横までで0個取得, 下のマスまでで0個取得, 下のマスまでで1個取得, 下のマスまでで2個取得, 下のマスまでで3個取得)
            dp[(r,c,0)]=max( dp[(r,c-1,0)], dp[(r-1,c,0)], dp[(r-1,c,1)], dp[(r-1,c,2)], dp[(r-1,c,3)] )
            #今のマスまでで1個取得=今のマスの左横までで1個取得
            dp[(r,c,1)]=dp[(r,c-1,1)]
            #今のマスまでで2個取得=今のマスの左横までで2個取得
            dp[(r,c,2)]=dp[(r,c-1,2)]
            #今のマスまでで3個取得=今のマスの左横までで3個取得
            dp[(r,c,3)]=dp[(r,c-1,3)]

print max(dp[(r,c,0)],dp[(r,c,1)],dp[(r,c,2)],dp[(r,c,3)])

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp=defaultdict(lambda: float("-inf")) 
#初期化
dp[(1,1,0)]=0
if value[(1,1)]!=0:
    dp[(1,1,1)]=value[(1,1)]


#1番下の行だけdpテーブルを埋める
r=1
for c in range(2,C+1):
    #今のマスにアイテムがある場合
    if 0<value[(r,c)]:
        #今のマスまでで0個取得=今のマスの左横までで0個取得
        dp[(r,c,0)]=dp[(r,c-1,0)]
        #今のマスまでで1個取得=max(今のマスの左横までで1個取得, 今のマスまでで0個取得+v)
        dp[(r,c,1)]=max( dp[(r,c-1,1)],dp[(r,c-1,0)]+value[(r,c)] )
        #今のマスまでで2個取得=max(今のマスの左横までで2個取得, 今のマスまでで1個取得+v)
        dp[(r,c,2)]=max( dp[(r,c-1,2)],dp[(r,c-1,1)]+value[(r,c)] )
        #今のマスまでで3個取得=max(今のマスの左横までで3個取得, 今のマスまでで2個取得+v)
        dp[(r,c,3)]=max( dp[(r,c-1,3)],dp[(r,c-1,2)]+value[(r,c)] )
    #今のマスにアイテムがない場合
    else:
        #左横のマスの値をスライド
        dp[(r,c,0)]=dp[(1,c-1,0)]
        dp[(r,c,1)]=dp[(r,c-1,1)]
        dp[(r,c,2)]=dp[(r,c-1,2)]
        dp[(r,c,3)]=dp[(r,c-1,3)]


#1番左の列のdpテーブルを埋める
c=1
for r in range(2,R+1):
    #今のマスにアイテムがある場合
    if 0<value[(r,c)]:
        #今のマスまでで0個取得=max(下のマスで0個取得、下のマスで1個取得)
        dp[(r,c,0)]=max(dp[(r-1,c,0)],dp[(r-1,c,1)])
        #今のマスまでで1個取得=max(下のマスで0個取得+v、下のマスで1個取得+v)
        dp[(r,c,1)]=max(dp[(r-1,c,0)]+value[(r,c)],dp[(r-1,c,1)]+value[(r,c)])
    #今のマスにアイテムがない場合
    else:
        #今のマスまでで0個取得=max(下のマスで0個取得、下のマスで1個取得)
        dp[(r,c,0)]=max(dp[(r-1,c,0)],dp[(r-1,c,1)])


for r in range(2,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[(r,c)]:
            #今のマスまでで0個取得=max(今のマスの左横までで0個取得, 下のマスまでで0個取得, 下のマスまでで1個取得, 下のマスまでで2個取得, 下のマスまでで3個取得)
            dp[(r,c,0)]=max( dp[(r,c-1,0)], dp[(r-1,c,0)], dp[(r-1,c,1)], dp[(r-1,c,2)], dp[(r-1,c,3)] )
            #今のマスまでで1個取得=max(今のマスの左横までで1個取得, 今のマスまでで0個取得+v, 下のマスまでで0個取得+v, 下のマスまでで1個取得+v, 下のマスまでで2個取得+v, 下のマスまでで3個取得+v)
            dp[(r,c,1)]=max( dp[(r,c-1,1)], dp[(r,c-1,0)]+value[(r,c)], dp[(r-1,c,0)]+value[(r,c)], dp[(r-1,c,1)]+value[(r,c)], dp[(r-1,c,2)]+value[(r,c)], dp[(r-1,c,3)]+value[(r,c)]  )
            #今のマスまでで2個取得=max(今のマスの左横までで2個取得, 今のマスまでで1個取得+v)
            dp[(r,c,2)]=max( dp[(r,c-1,2)], dp[(r,c-1,1)]+value[(r,c)] )
            #今のマスまでで3個取得=max(今のマスの左横までで3個取得, 今のマスまでで2個取得+v)
            dp[(r,c,3)]=max( dp[(r,c-1,3)],dp[(r,c-1,2)]+value[(r,c)] )
        #今のマスにアイテムがない場合
        else:
            #今のマスまでで0個取得=max(今のマスの左横までで0個取得, 下のマスまでで0個取得, 下のマスまでで1個取得, 下のマスまでで2個取得, 下のマスまでで3個取得)
            dp[(r,c,0)]=max( dp[(r,c-1,0)], dp[(r-1,c,0)], dp[(r-1,c,1)], dp[(r-1,c,2)], dp[(r-1,c,3)] )
            #今のマスまでで1個取得=今のマスの左横までで1個取得
            dp[(r,c,1)]=dp[(r,c-1,1)]
            #今のマスまでで2個取得=今のマスの左横までで2個取得
            dp[(r,c,2)]=dp[(r,c-1,2)]
            #今のマスまでで3個取得=今のマスの左横までで3個取得
            dp[(r,c,3)]=dp[(r,c-1,3)]

print max(dp[(r,c,0)],dp[(r,c,1)],dp[(r,c,2)],dp[(r,c,3)])

Submission status: Wrong Answer

Solution code:
# -*- coding: utf-8 -*-
import sys
from collections import defaultdict
R,C,K=map(int, sys.stdin.readline().split())

value=defaultdict(lambda: 0)
for _ in range(K):
    r,c,v=map(int, sys.stdin.readline().split())
    value[(r,c)]=v

dp=defaultdict(lambda: float("-inf")) 
#初期化
dp[(1,1,0)]=0
dp[(1,1,1)]=value[(1,1)]


#1番下の行だけdpテーブルを埋める
r=1
for c in range(2,C+1):
    #今のマスにアイテムがある場合
    if 0<value[(r,c)]:
        #今のマスまでで0個取得=今のマスの左横までで0個取得
        dp[(r,c,0)]=dp[(r,c-1,0)]
        #今のマスまでで1個取得=max(今のマスの左横までで1個取得, 今のマスまでで0個取得+v)
        dp[(r,c,1)]=max( dp[(r,c-1,1)],dp[(r,c-1,0)]+value[(r,c)] )
        #今のマスまでで2個取得=max(今のマスの左横までで2個取得, 今のマスまでで1個取得+v)
        dp[(r,c,2)]=max( dp[(r,c-1,2)],dp[(r,c-1,1)]+value[(r,c)] )
        #今のマスまでで3個取得=max(今のマスの左横までで3個取得, 今のマスまでで2個取得+v)
        dp[(r,c,3)]=max( dp[(r,c-1,3)],dp[(r,c-1,2)]+value[(r,c)] )
    #今のマスにアイテムがない場合
    else:
        #左横のマスの値をスライド
        dp[(r,c,0)]=dp[(1,c-1,0)]
        dp[(r,c,1)]=dp[(r,c-1,1)]
        dp[(r,c,2)]=dp[(r,c-1,2)]
        dp[(r,c,3)]=dp[(r,c-1,3)]


#1番左の列のdpテーブルを埋める
c=1
for r in range(2,R+1):
    #今のマスにアイテムがある場合
    if 0<value[(r,c)]:
        #今のマスまでで0個取得=max(下のマスで0個取得、下のマスで1個取得)
        dp[(r,c,0)]=max(dp[(r-1,c,0)],dp[(r-1,c,1)])
        #今のマスまでで1個取得=max(下のマスで0個取得+v、下のマスで1個取得+v)
        dp[(r,c,1)]=max(dp[(r-1,c,0)]+value[(r,c)],dp[(r-1,c,1)]+value[(r,c)])
    #今のマスにアイテムがない場合
    else:
        #今のマスまでで0個取得=max(下のマスで0個取得、下のマスで1個取得)
        dp[(r,c,0)]=max(dp[(r-1,c,0)],dp[(r-1,c,1)])


for r in range(2,R+1):
    for c in range(2,C+1):
        #今のマスにアイテムがある場合
        if 0<value[(r,c)]:
            #今のマスまでで0個取得=max(今のマスの左横までで0個取得, 下のマスまでで0個取得, 下のマスまでで1個取得, 下のマスまでで2個取得, 下のマスまでで3個取得)
            dp[(r,c,0)]=max( dp[(r,c-1,0)], dp[(r-1,c,0)], dp[(r-1,c,1)], dp[(r-1,c,2)], dp[(r-1,c,3)] )
            #今のマスまでで1個取得=max(今のマスの左横までで1個取得, 今のマスまでで0個取得+v, 下のマスまでで0個取得+v, 下のマスまでで1個取得+v, 下のマスまでで2個取得+v, 下のマスまでで3個取得+v)
            dp[(r,c,1)]=max( dp[(r,c-1,1)], dp[(r,c-1,0)]+value[(r,c)], dp[(r-1,c,0)]+value[(r,c)], dp[(r-1,c,1)]+value[(r,c)], dp[(r-1,c,2)]+value[(r,c)], dp[(r-1,c,3)]+value[(r,c)]  )
            #今のマスまでで2個取得=max(今のマスの左横までで2個取得, 今のマスまでで1個取得+v)
            dp[(r,c,2)]=max( dp[(r,c-1,2)], dp[(r,c-1,1)]+value[(r,c)] )
            #今のマスまでで3個取得=max(今のマスの左横までで3個取得, 今のマスまでで2個取得+v)
            dp[(r,c,3)]=max( dp[(r,c-1,3)],dp[(r,c-1,2)]+value[(r,c)] )
        #今のマスにアイテムがない場合
        else:
            #今のマスまでで0個取得=max(今のマスの左横までで0個取得, 下のマスまでで0個取得, 下のマスまでで1個取得, 下のマスまでで2個取得, 下のマスまでで3個取得)
            dp[(r,c,0)]=max( dp[(r,c-1,0)], dp[(r-1,c,0)], dp[(r-1,c,1)], dp[(r-1,c,2)], dp[(r-1,c,3)] )
            #今のマスまでで1個取得=今のマスの左横までで1個取得
            dp[(r,c,1)]=dp[(r,c-1,1)]
            #今のマスまでで2個取得=今のマスの左横までで2個取得
            dp[(r,c,2)]=dp[(r,c-1,2)]
            #今のマスまでで3個取得=今のマスの左横までで3個取得
            dp[(r,c,3)]=dp[(r,c-1,3)]

print max(dp[(r,c,0)],dp[(r,c,1)],dp[(r,c,2)],dp[(r,c,3)])

Submission status: Wrong Answer

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

dp = [[[0]*4 for _ in range(C)] for _ in range(R)]
dp[0][0][1] = meiro[0][0]
#dp[r][c][i]:=(r, c)にいてr行上のアイテムをi個持っているときの、その価値の総和の最大値
for r in range(R):
    for c in range(C):
        if r < R-1:
            dp[r+1][c][0] = max(dp[r+1][c][0], dp[r][c][1], dp[r][c][2], dp[r][c][3])
            dp[r+1][c][1] = max(dp[r][c][1],
                                dp[r][c][1] + meiro[r+1][c],
                                dp[r][c][2],
                                dp[r][c][2] + meiro[r+1][c],
                                dp[r][c][3],
                                dp[r][c][3] + meiro[r+1][c],
                                dp[r][c][0] + meiro[r+1][c])
        if c < C-1:
            dp[r][c+1][1] = max(dp[r][c+1][1],
                                dp[r][c][1],
                                dp[r][c][0] + meiro[r][c+1])
            dp[r][c+1][2] = max(dp[r][c+1][2],
                                dp[r][c][2],
                                dp[r][c][1] + meiro[r][c+1])
            dp[r][c+1][3] = max(dp[r][c+1][3],
                                dp[r][c][2],
                                dp[r][c][2] + meiro[r][c+1])
print(max(dp[R-1][C-1]))

Submission status: Wrong Answer

Solution code:
import heapq
R, C, K = map(int, input().split())
meiro = [[0]*C for _ in range(R)]
for _ in range(K):
    r, c, v = map(int, input().split())
    r -= 1
    c -= 1
    meiro[r][c] = v

dp = [[[0]*4 for _ in range(C)] for _ in range(R)]
dp[0][0][1] = meiro[0][0]
#dp[r][c][i]:=(r, c)にいてr行上のアイテムをi個持っているときの、その価値の総和の最大値
for r in range(R):
    for c in range(C):
        if r-1 >= 0:
            dp[r][c][0] = max(dp[r][c][0], dp[r-1][c][1], dp[r-1][c][2], dp[r-1][c][3])
            dp[r][c][1] = max(dp[r-1][c][1],
                              dp[r-1][c][1] + meiro[r][c],
                              dp[r-1][c][2],
                              dp[r-1][c][2] + meiro[r][c],
                              dp[r-1][c][3],
                              dp[r-1][c][3] + meiro[r][c],
                              dp[r-1][c][0] + meiro[r][c])
        if c-1 >= 0:
            dp[r][c][1] = max(dp[r][c][1],
                              dp[r][c-1][1],
                              dp[r][c-1][0] + meiro[r][c])
            dp[r][c][2] = max(dp[r][c][2],
                              dp[r][c - 1][2],
                              dp[r][c - 1][1] + meiro[r][c])
            dp[r][c][3] = max(dp[r][c][3],
                              dp[r][c-1][2],
                              dp[r][c-1][2] + meiro[r][c])
print(max(dp[R-1][C-1]))