본문 바로가기
Programming/Programmers

[프로그래머스] 도둑질(Python)

by 데이터현 2021. 8. 12.

https://programmers.co.kr/learn/courses/30/lessons/42897

 

코딩테스트 연습 - 도둑질

도둑이 어느 마을을 털 계획을 하고 있습니다. 이 마을의 모든 집들은 아래 그림과 같이 동그랗게 배치되어 있습니다. 각 집들은 서로 인접한 집들과 방범장치가 연결되어 있기 때문에 인접한

programmers.co.kr

def solution(money):
    # dp[i] = max(dp[i-1], dp[i-2]+money[i]) 점화식
    dp = [0]*len(money)
    #첫번째 집을 터는 경우
    dp[0]=money[0]
    dp[1]=money[0]
    for i in range(2,len(money)-1): #마지막 집은 털면 안됨
        dp[i] = max(dp[i-1], dp[i-2]+money[i])
    firstmoney = max(dp)
    #첫번째 집을 안터는 경우
    dp = [0]*len(money)
    dp[1]=money[1]
    for i in range(2,len(money)): #마지막 집 털기 가능
        dp[i] = max(dp[i-1], dp[i-2]+money[i])
    return max(firstmoney,max(dp))

댓글