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

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

프로그래머스

코딩테스트 연습 > Summer/Winter Coding(~2018) > 스티커 모으기(2)

 

문제 설명

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

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

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

 

제한 사항

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

 

문제 접근

 

첫 번째 원소를 선택한다고 하면 마지막 원소를 선택 못하는 게 이번 문제의 핵심이다.

또한 원소의 개수가 3 이하면 3개 중 제일 큰 값이 정답이다. 다른 선택 사항은 없다.

 

DP1 : 첫 번째 원소를 사용하는 누적합 

DP2 : 첫 번째 원소를 사용하지 않는 누적합 즉 마지막 원소인 10을 선택하는게 가능한 case

 

[0] : 이번 원소를 선택하는 case

[1] : 이번 원소를 선택하지 않는 case

 

[0] : 이번 원소 sticker[i - 1] + 이전 DP에서 선택 가능한 값 중에서 max  

[1] : 이전 DP에서 선택 가능한 값 중에서 max 

 

마지막 원소인 경우에는 DP2 (첫 번째 원소를 사용하지 않는 DP 에서 최댓값을 구한다.)

  

정답 코드

class Solution {
    public int solution(int sticker[]) {  
        int size = sticker.length;
        int max = 0;

        int result = 0;
        
        for(int i = 0; i < size; i++){
            result = Math.max(result, sticker[i]);
        }
        
        if(size <= 3) return result;
        
        // 첫 번째 원소를 사용하는 
        int [][] DP1  = new int [size+1][2];  // 0은 이번 원소를 사용 1은 이번 원소 사용 x
        // 첫 번째 원소를 사용하지 않는 
        int [][] DP2  = new int [size+1][2]; 
        
        DP1[1][0] = sticker[0];
        DP2[2][0] = sticker[1];
        
        for(int i = 3; i <= size; i++){
            if(i != size){
                DP2[i][0] = Math.max(DP2[i-1][1], Math.max(DP2[i-2][0], DP2[i-2][1])) + sticker[i-1];
                DP2[i][1] = Math.max(DP2[i-1][1], Math.max(DP2[i-1][0], Math.max(DP2[i-2][0], DP2[i-2][1])));
                max = Math.max(DP2[i][0], Math.max(max, DP2[i][1]));
                
                DP1[i][0] = Math.max(DP1[i-1][1], Math.max(DP1[i-2][0], DP1[i-2][1])) + sticker[i-1];
                DP1[i][1] = Math.max(DP1[i-1][1], Math.max(DP1[i-1][0], Math.max(DP1[i-2][0], DP1[i-2][1])));
                max = Math.max(DP1[i][0], Math.max(max, DP1[i][1]));
            }
            else{
                DP2[i][0] = Math.max(DP2[i-1][1], Math.max(DP2[i-2][0], DP2[i-2][1])) + sticker[i-1];
                DP2[i][1] = Math.max(DP2[i-1][1], Math.max(DP2[i-1][0], Math.max(DP2[i-2][0], DP2[i-2][1])));
                max = Math.max(DP2[i][0], Math.max(max, DP2[i][1]));
            }
        }
        
        return max;
    }
}