몸과 마음이 건전한 SW 개발자

프로그래머스 [Lv. 3] 스티커 모으기 {언어 : Python} 본문

알고리즘

프로그래머스 [Lv. 3] 스티커 모으기 {언어 : Python}

스위태니 2024. 6. 14. 01:23

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/12971

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

정답 코드

def dynamicProgramming(length, firstUsed, sticker):
    dp = [0 for _ in range(length)]
    maxV = 0
    if firstUsed:
        dp[0] = sticker[0]
        dp[1] = sticker[0]
        length -= 1
    else:
        dp[1] = sticker[1]
    for i in range(2, length):
        value = max(dp[i-2]+sticker[i], dp[i-1])
        dp[i] = value
        if maxV < value:
            maxV = value
    return maxV

def solution(sticker):
    lenS = len(sticker)
    if lenS == 1:
        return sticker[0]
    elif lenS == 2:
        return max(sticker)
    answer = max(dynamicProgramming(lenS, True, sticker), dynamicProgramming(lenS, False, sticker))
        
    return answer

풀이 방법

  1. 첫 번째 스티커를 떼냐 안떼냐로 나눈다.
    1. 스티커를 떼는 경우 dp의 0 번째 인덱스와 1번째 인덱스는 당연히 sticker[0]이 된다.
    2. 하지만 떼지 않는 경우 dp[0]은 0이며 dp[1]이 sticker[1]이 된다.
  2. 길이가 1인 경우와 2인 경우는 계산할 필요가 없다.
  3. dp를 구하는 과정에서 최대값을 return하고 최대값끼리 비교해서 answer에 치환한다.
  4. answer를 반환하면 끝!

느낀점

  • 오랜만에 해서 감이 안잡혔다.
  • 여러번 풀어도 결국 시간이 지나면 까먹는다.
  • 기록을 통해 남긴다.