본문 바로가기
Python/알고리즘문제

[프로그래머스][동적계획법] 도둑질

by 붕어사랑 티스토리 2021. 4. 14.
반응형

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

 

프로그래머스

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

programmers.co.kr

문제 해결 아이디어

  • 어차피 돈은 양수이기에 두칸씩 띄어서 선택하면 장땡이다
  • 첫번째 집과 마지막집은 붙어있기에 동시에 선택을 못한다
  • 둘중하나 선택을 안하면 직선처럼 문제를 생각할 수 있다
  • 두케이스중 큰값을 골라준다

 

def solution(money):
    answer = 0
    # 마지막집을 선택하지 않은경우
    dp = [0] * len(money)
    dp[0] = money[0]
    dp[1] = max(money[0], money[1])
    for i in range(2, len(money) - 1):
        dp[i] = max(dp[i-2] + money[i], dp[i-1])
    answer = max(answer, max(dp))

    #첫번째 집을 선택하지 않은 경우
    dp = [0] * len(money)
    dp[0] = 0
    dp[1] = money[1]
    for i in range(2, len(money)):
        dp[i] = max(dp[i-2] + money[i], dp[i-1])
    answer = max(answer, max(dp))
    
    return answer
반응형

댓글