[프로그래머스] 도둑질

Turtle·2024년 8월 29일
0
post-thumbnail

🗃️문제 설명

도둑이 어느 마을을 털 계획을 하고 있습니다. 이 마을의 모든 집들은 아래 그림과 같이 동그랗게 배치되어 있습니다.

각 집들은 서로 인접한 집들과 방범장치가 연결되어 있기 때문에 인접한 두 집을 털면 경보가 울립니다.

각 집에 있는 돈이 담긴 배열 money가 주어질 때, 도둑이 훔칠 수 있는 돈의 최댓값을 return 하도록 solution 함수를 작성하세요.

🖥️코드

def solution(money):
    # 원형으로 되어있기 때문에 만약 첫 번째 집부터 턴다고 가정한다면
    # 마지막 집은 털 수 없다.
    dp1 = [0] * len(money)
    dp1[0] = money[0]
    dp1[1] = money[0]
    for i in range(2, len(money)-1):
        dp1[i] = max(dp1[i-1], money[i] + dp1[i-2])
        
    # 원형으로 되어있기 때문에 만약 두 번째 집부터 턴다고 가정한다면
    # 마지막 집을 털 수 있다.
    dp2 = [0] * len(money)
    dp2[1] = money[1]
    for i in range(2, len(money)):
        dp2[i] = max(dp2[i-1], money[i] + dp2[i-2])
    
    return max(max(dp1), max(dp2))

🔒문제 출처

프로그래머스 - 도둑질

0개의 댓글