스티커 모으기

문제 ✨

Summer/Winter Coding(~2018) > 스티커 모으기

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

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

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

def solution(sticker):
    if len(sticker)==1: return sticker[0]
    zero, end = [0]*len(sticker), [0]*len(sticker)
    
    zero[0], zero[1] = sticker[0], max(sticker[0], sticker[1])
    end[0], end[1] = 0, sticker[1]
    
    for i in range(2, len(sticker)-1):
        zero[i] = max(zero[i-1], zero[i-2]+sticker[i])
    
    for i in range(2, len(sticker)):
        end[i] = max(end[i-1], end[i-2]+sticker[i])

    return max(max(zero), max(end))

처음엔 짝수와 홀수만 더하면 해결되는 줄 알았는데 짝수와 홀수가 섞인 1번째와 4번째를 더한 결과가 더 클 수 있는 경우가 존재
0번째 스티커와 마지막 스티커는 같이 사용될 수 없으므로 둘 중 0번째만 포함한 배열과 마지막만 포함한 배열로 2번 계산