[프로그래머스/Python] 스티커 모으기(2)

문제 정보

제목 : 스티커 모으기(2)

난이도 : Lv.3

사용 언어 : Python

문제 링크

https://school.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
6, 11, 9, 10이 적힌 스티커를 떼어 냈을 때 36으로 최대가 됩니다.

입출력 예 #2
3, 5가 적힌 스티커를 떼어 냈을 때 8로 최대가 됩니다.

---

나의 풀이

스티커의 길이가 1이라면 해당 스티커를 return 합니다. 아니라면 첫 번째 스티커를 뜯은 경우와, 첫 번째 스티커를 뜯지 않은 경우로 나누어 구합니다. 첫 번째 스티커를 뜯은 경우에는 dp1의 초기값을 sticker[0]으로 저장하고, 2부터 sticker 길이 -1까지 반복하는 for루프를 이용해 dp1의 값을 계산합니다. 첫 번째 스티커를 뜯지 않은 경우를 구하는 것은 dp2의 초기값을 sticker[1]로 저장하고, 2부터 sticker 길이까지 반복하는 for 루프를 이용해 dp2의 값을 계산합니다. answer에는 dp1과 dp2의 최대값 중 더 큰 수를 저장하여 return합니다.

 

코드

def solution(sticker):
    answer = []
    if( len(sticker)==1 ): # 스티커 길이가 1인 경우
        return sticker[0]
    
    # 첫 번째 스티커를 뜯은 경우
    dp1 = [0] * len(sticker)
    dp1[0] = sticker[0]
    dp1[1] = dp1[0]
    for i in range(2, len(sticker)-1):
        dp1[i] = max(dp1[i-2] + sticker[i], dp1[i-1])
    
    # 첫 번째 스티커를 뜯지 않은 경우
    dp2 = [0]*len(sticker)
    dp2[1] = sticker[1]
    for i in range(2,len(sticker)):
        dp2[i] = max( dp2[i-2] + sticker[i],dp2[i-1] )
    answer = max( max(dp1), max(dp2))
    return answer