프로그래머스 Summer/Winter Coding(~2018) - 스티커 모으기(2)

이환희·2021년 4월 27일
0

Algorithm

목록 보기
11/47

문제 설명

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

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

제한 사항

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

풀이

def solution(sticker):
    answer = 0
    if len(sticker) == 1:
        return sticker[0]
    dp1 = [0 for _ in range(len(sticker))]  # dp1은 첫번째를 쓸때
    dp2 = [0 for _ in range(len(sticker))]  # dp2은 첫번째를 안 쓸때
    dp1[0], dp1[1] = sticker[0], sticker[0]  # 첫번째를 쓰므로 두번째도 첫번째랑 같음
    dp2[0], dp2[1] = 0, sticker[1]  # 첫번째를 안 쓰므로 맨처음은 0 이고 두번째 부터 시작
    for i in range(2, len(sticker)-1):  # 첫번째를 쓰면 맨 마지막을 제외하고 dp계산
        # 스티커의 i번째 쓸때와 안 쓸때중 큰값을 저장
        dp1[i] = max(dp1[i-2]+sticker[i], dp1[i-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

1차원 dp문제
첫번째 스티커를 쓸때와 안 쓸때를 구분하여 구함

첫번째 스티커를 쓸때

  • 첫번째 스티커를 쓰면 dp1의 첫번째와 두번째 모두 첫번째 스티커값으로 채워짐
  • 맨 마지막이 첫번째와 붙어있으므로 dp1의 마지막을 제외하고 계산
  • 점화식 : dp1[i] = max(dp1[i-2]+sticker[i], dp1[i-1])
  • i번째를 쓸때 dp1[i-2]+sticker[i] 와 안 쓸때 dp1[i-1] 중 더 큰값을 기록

첫번째 스티커를 안 쓸때

  • 안 쓰면 dp2의 첫번째는 0, 두번째는 스티커의 두번째 값으로 채워짐
  • 맨 마지막까지 계산
  • 점화식은 위와 동일

dp1과 dp2에서 가장 큰 값끼리 비교해서 더 큰 값 리턴

0개의 댓글

관련 채용 정보