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

출처

🌐 코딩테스트 연습 - 스티커 모으기(2) | 프로그래머스 스쿨

문제

문제 설명

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

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

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

제한사항

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

입출력 예시

sticker answer

[14, 6, 5, 11, 3, 9, 2, 10]	36
[1, 3, 2, 5, 4]	8

풀이

#include <vector>
#include <algorithm>
using namespace std;

int solution(vector<int> sticker)
{    
    int n = sticker.size();
    if (n == 1) return sticker[0];
    if (n == 2) return max(sticker[0], sticker[1]);
    if (n == 3) return max({sticker[0], sticker[1], sticker[2]});
    
    vector<int> v(100000,0);
    v[0] = sticker[0];
    v[1] = sticker[1];
    v[2] = v[0] + sticker[2];
    for(int i = 3; i < sticker.size()-1; i++)
    {        
        int m = max(v[i - 2], v[i - 3]);
        v[i] = m + sticker[i];
    }
    
    vector<int> v1(100000,0);
    v1[0] = sticker[sticker.size()-1];
    v1[1] = sticker[1] + v1[0];
    v1[2] = v1[0] + sticker[2];

    for(int i = 2; i < sticker.size()-2; i++)
    {        
        int m = max(v1[i - 2], v1[i - 3]);
        v1[i] = m + sticker[i];
    }

    
    return max(*max_element(v.begin(), v.end()), *max_element(v1.begin(), v1.end()));
}

Review

1️⃣ 한꺼번에 처리하려다 오래 걸렸다… 따로따로 처리가 훨씬 나을 수도

2️⃣size가 1~3일 때, 예외 처리하지 못했다