문제 설명:
N개의 스티커가 원형으로 연결되어 있습니다. 다음 그림은 N = 8인 경우의 예시입니다.
원형으로 연결된 스티커에서 몇 장의 스티커를 뜯어내어 뜯어낸 스티커에 적힌 숫자의 합이 최대가 되도록 하고 싶습니다. 단 스티커 한 장을 뜯어내면 양쪽으로 인접해있는 스티커는 찢어져서 사용할 수 없게 됩니다. 예를 들어 위 그림에서 14가 적힌 스티커를 뜯으면 인접해있는 10, 6이 적힌 스티커는 사용할 수 없습니다. 스티커에 적힌 숫자가 배열 형태로 주어질 때, 스티커를 뜯어내어 얻을 수 있는 숫자의 합의 최댓값을 return 하는 solution 함수를 완성해 주세요. 원형의 스티커 모양을 위해 배열의 첫 번째 원소와 마지막 원소가 서로 연결되어 있다고 간주합니다.
제한사항 sticker는 원형으로 연결된 스티커의 각 칸에 적힌 숫자가 순서대로 들어있는 배열로, 길이(N)는 1 이상 100,000 이하입니다. sticker의 각 원소는 스티커의 각 칸에 적힌 숫자이며, 각 칸에 적힌 숫자는 1 이상 100 이하의 자연수입니다. 원형의 스티커 모양을 위해 sticker 배열의 첫 번째 원소와 마지막 원소가 서로 연결되어있다고 간주합니다.
입출력 예 #1
- 6, 11, 9, 10이 적힌 스티커를 떼어 냈을 때 36으로 최대가 됩니다.
입출력 예 #2
- 3, 5가 적힌 스티커를 떼어 냈을 때 8로 최대가 됩니다.
def solution(sticker):
if len(sticker) <= 2:
return max(sticker)
dp = [0] * len(sticker)
no_dp = [0] * len(sticker)
dp[0] = sticker[0]
dp[1] = dp[0]
# 맨 앞 스티커 뜯은 경우 >> 끝에꺼는 못 뜯음 -> dp[-2] 최대
for i in range(2, len(sticker) - 1):
dp[i] = max(dp[i - 2] + sticker[i], dp[i - 1])
# 맨 앞 스티커 안 뜯은경우 > 끝에꺼 뜯음 > no_dp[-1]
for i in range(1, len(sticker)):
no_dp[i] = max(no_dp[i - 2] + sticker[i], no_dp[i - 1])
return max(no_dp[-1], dp[-2])
맨 앞에 나온 스티커를 뜯었는지 , 안 뜯었는지를 기준으로 dp 두 개를 만들어준다.
첫 번째 스티커를 뜯은 dp에 경우 맨 마지막 인덱스의 스티커는 무조건 뜯을 수 없으므로, dp[-2] 가 최댓값이고
첫 번째 스티커를 안 뜯은 dp는 맨 마지막 스티커를 무조건 뜯을 수 있으므로 dp[-1] 가 최댓값이 된다.
댓글