알고리즘 공부/DP

[프로그래머스] 스티커 모으기(2)

kdhoooon 2021. 6. 21. 23:11

문제


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


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

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

 

 

[제한 사항]

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

 

 

 

 

 

 

풀이


우선 2가지 경우의 수를 고려해줘야 한다.

  1.  첫 번째 카드를 뽑는다.
  2.  첫 번째 카드를 뽑지 않는다.

이렇게 두가지 경우의 수를 고려하여 다음 DP를 이용하여 풀 수 있다.

현재의 스티커 와 두 개전의 스티커의 합이 한 개전의 스티커보다 크면 둘다 뽑는다.

Sticker[i] + Sticker[i -2] > Sticker[i - 1] 

위 식이 성립하면 현재의 스티커와 2개 전의 스티커를 뽑아야한다.

 

따라서 다음과 같은 식을 도출 할 수 있다.

DP[i] = Math.max(DP[i-1], DP[i-2] + sticker[i])

 

2가지 경우의 수를 중 더 큰값을 뽑아주면 된다.

 

<전체코드>

class Solution {
    public int solution(int sticker[]) {
        int answer = 0;
        
        if(sticker.length == 1){
            return sticker[0];
        }
        
        int[] dp1 = new int[sticker.length];
        int[] dp2 = new int[sticker.length];
        
        dp1[0] = 0;
        dp1[1] = sticker[1];
        for(int i = 2 ; i < sticker.length ; i++){
            dp1[i] = Math.max(dp1[i-1], dp1[i - 2] + sticker[i]);
        }
        
        dp2[0] = sticker[0];
        dp2[1] = sticker[0];
        for(int i = 2 ; i < sticker.length ; i++){
            dp2[i] = Math.max(dp2[i-1], dp2[i - 2] + sticker[i]);
        }
        
        answer = Math.max(dp1[sticker.length - 1], dp2[sticker.length - 2]);
        
        return answer;
    }
}

'알고리즘 공부 > DP' 카테고리의 다른 글

[백준] 11062 카드 게임  (0) 2021.08.07
[프로그래머스] 도둑질  (0) 2021.07.29
[프로그래머스] 멀리 뛰기  (0) 2021.06.09
[프로그래머스] 거스름돈  (0) 2021.06.09
[프로그래머스] 등굣길  (0) 2021.05.23