[Python/Programmers] 스티커 모으기(2)

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

프로그래머스

문제 설명

N개의 스티커가 원형으로 연결되어 있습니다. 다음 그림은 N = 8인 경우의 예시입니다.

원형으로 연결된 스티커에서 몇 장의 스티커를 뜯어내어 뜯어낸 스티커에 적힌 숫자의 합이 최대가 되도록 하고 싶습니다. 단 스티커 한 장을 뜯어내면 양쪽으로 인접해있는 스티커는 찢어져서 사용할 수 없게 됩니다.

예를 들어 위 그림에서 14가 적힌 스티커를 뜯으면 인접해있는 10, 6이 적힌 스티커는 사용할 수 없습니다. 스티커에 적힌 숫자가 배열 형태로 주어질 때, 스티커를 뜯어내어 얻을 수 있는 숫자의 합의 최댓값을 return 하는 solution 함수를 완성해 주세요. 원형의 스티커 모양을 위해 배열의 첫 번째 원소와 마지막 원소가 서로 연결되어 있다고 간주합니다.

풀이

Code

def solution(sticker):
    # 1. 스티커가 1개인 경우
    if len(sticker) == 1 : return sticker[0]
    # 2. DP 생성
    dp1 = [0] + sticker[:-1]
    dp2 = [0] + sticker[1:]
    # 3.
    for i in range(2, len(dp1)) :
        # 3-1. 0번 인덱스의 스티커를 뜯지 않은 dp 처리
        dp1[i] = max(dp1[i-2] + dp1[i], dp1[i-1])
        # 3-2. 0번 인덱스의 스티커를 뜯은 dp 처리
        dp2[i] = max(dp2[i-2] + dp2[i], dp2[i-1])
    # 4. 최댓값 출력
    return max(dp1[-1], dp2[-1])