몸과 마음이 건전한 SW 개발자

프로그래머스 [Lv. 2] 땅따먹기 {언어 : Python} 본문

알고리즘

프로그래머스 [Lv. 2] 땅따먹기 {언어 : Python}

스위태니 2024. 5. 20. 20:22

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/12913?language=python3

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

정답 코드

def solution(land):
    lenL = len(land)
    dp = [[0 for _ in range(4)] for _ in range(lenL)]
    dp[0] = land[0]
    for i in range(1, lenL):
        for j in range(4):
            dp[i][j] = max(dp[i-1][(j+1)%4], dp[i-1][(j+2)%4], dp[i-1][(j+3)%4]) + land[i][j]
    answer = max(dp[-1])
    return answer

풀이 방법

  1. dp[i][j]에 j열을 제외하고 나머지 열에서 최대값과 현재 값을 더해가며 마지막 행까지 dp를 구한다.

느낀점

  • 오랜만에 dp 쉬운 문제를 풀어서 좋다.