220309 - 스티커 모으기(2)

◾ 스티커 모으기(2) : 프로그래머스 LEVEL 3

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

문제

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

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

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

---

입력

• sticker는 원형으로 연결된 스티커의 각 칸에 적힌 숫자가 순서대로 들어있는 배열로, 길이(N)는 1 이상 100,000 이하입니다.
• sticker의 각 원소는 스티커의 각 칸에 적힌 숫자이며, 각 칸에 적힌 숫자는 1 이상 100 이하의 자연수입니다.
• 원형의 스티커 모양을 위해 sticker 배열의 첫 번째 원소와 마지막 원소가 서로 연결되어있다고 간주합니다.

출력

• 스티커를 뜯어내어 얻을 수 있는 수자의 합의 최대값

입출력 예

sticker	answer
[14, 6, 5, 11, 3, 9, 2, 10]	36
[1, 3, 2, 5, 4]	8

---

◾ 풀이

1. 해설

• 전체를 확인하면 시간초과가 발생하므로 동적 계획법을 이용해 해결한다.
• 첫 번째 스티커를 뜯는 경우와 두 번쨰 스티커를 뜯는 경우 2가지 경우가 존재한다.
  • 첫 번째 스티커를 뜯을 경우 마지막 스티커를 제외해야 하므로 두 가지 경우를 따로 확인하고 최대값을 비교한다.

---

2. 프로그램

• dp[i] = max(dp[i-1], dp[i-2] + sticker[i])

1. 첫 번째 스티커를 뜯는 경우 탐색(0 ~ 마지막 인덱스-1)
2. 두 번째 스티커를 뜯는 경우 탐색(0 ~ 마지막 인덱스)
3. 각각의 최대값 비교 후 반환

# 코드
def solution(sticker):
    answer = 0

    if len(sticker) == 1:
        return sticker[0]
    # 첫 번째 스티커를 뜯는 경우
    # 마지막 스티커 제외
    dp = [0] * len(sticker)
    dp[0], dp[1] = sticker[0], sticker[0]
    for i in range(2, len(sticker) - 1):
        dp[i] = max(dp[i-1], dp[i-2] + sticker[i])
    answer = max(dp)

    # 두 번째 스티커를 뜯는 경우
    # 마지막 스티커까지 적용
    dp = [0] * len(sticker)
    dp[0], dp[1] = 0, sticker[1]
    for i in range(2, len(sticker)):
        dp[i] = max(dp[i-1], dp[i-2] + sticker[i])
    answer = max(answer, max(dp))

    return answer