스티커 모으기(2)
원형으로 연결된 스티커에서 몇 장의 스티커를 뜯어내어 뜯어낸 스티커에 적힌 숫자의 합이 최대가 되도록 하고 싶습니다. 단 스티커 한 장을 뜯어내면 양쪽으로 인접해있는 스티커는 찢어져서 사용할 수 없게 됩니다.
예를 들어 위 그림에서 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로 최대가 됩니다.
pick or not pick => 전형적인 DP 문제
def solution(sticker):
answer = 0
n = len(sticker)
if n <=3 :
return max(sticker)
dp1 = n * [0] # 첫번째
dp2 = n * [0] # 첫번째 x
# 초기화 셋팅
dp1[0] = sticker[0] # 첫번째 선택
dp1[1] = dp1[0] # 첫번째 = 두번째
dp2[0] = 0 # 첫번째 선택 안하니까 0
dp2[1] = sticker[1] # 선택하니까
for i in range(2,n-1): # 첫번째 pick => 세번째~ 마지막 전까지 가능
dp1[i] = max(sticker[i] + dp1[i-2],dp1[i-1])
for i in range(2,n): #두 번째 스티커를 선택 가능, 그러나 두 번째 스티커의 값 dp2[1]에 할당
# 따라서 3번째인 2부터 시작
dp2[i] = max(sticker[i] + dp2[i-2] ,dp2[i-1])
return max(dp1[-2], dp2[-1])
#dp는 i까지 구할 수 있는 최대의 합이 저장