알고리즘
프로그래머스 [Lv. 2] 땅따먹기 {언어 : Python}
스위태니
2024. 5. 20. 20:22
728x90
문제 링크
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
풀이 방법
- dp[i][j]에 j열을 제외하고 나머지 열에서 최대값과 현재 값을 더해가며 마지막 행까지 dp를 구한다.
느낀점
- 오랜만에 dp 쉬운 문제를 풀어서 좋다.
728x90