[프로그래머스]땅따먹기

https://school.programmers.co.kr/learn/courses/30/lessons/12913

프로그래머스

---

def solution(land):

    n = len(land)
    
    dp = [[0 for j in range(4)] for i in range(n)]
    
    for i in range(4):
        dp[0][i] = land[0][i]
    for i in range(1, n):
        dp[i][0] = land[i][0] + max(dp[i-1][1], dp[i-1][2], dp[i-1][3])
        dp[i][1] = land[i][1] + max(dp[i-1][0], dp[i-1][2], dp[i-1][3]) 
        dp[i][2] = land[i][2] + max(dp[i-1][0], dp[i-1][1], dp[i-1][3]) 
        dp[i][3] = land[i][3] + max(dp[i-1][0], dp[i-1][1], dp[i-1][2]) 
    
    return max(dp[n-1])

일단 이렇게 제출했는데

for문 내부에 하드코딩한 점이 너무 마음에 걸렸다.

 

 

다른 분들 풀이를 보니까 이렇게 풀면 되는 거였다.

보고 배워야겠따. 이러려고 파이썬 하는 거지

 land[i][j] = land[i][j] + max(land[i-1][:j]+land[i-1][j+1:])