[프로그래머스/C++] 풍선 터트리기

미숙한 블로그 주인이 코딩테스트 문제를 풀어가는 과정을 담은 글입니다. 이 풀이가 효율적인 풀이가 아닐 수 있으며, 부정확한 정보가 많이 있을 수 있습니다. 보완해야할 점이 있다면 댓글로 남겨주세요!

 

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

프로그래머스

풍선 터트리기

문제

일렬로 나열된 n개의 풍선이 있습니다. 모든 풍선에는 서로 다른 숫자가 써져 있습니다. 당신은 다음 과정을 반복하면서 풍선들을 단 1개만 남을 때까지 계속 터트리려고 합니다.

1. 임의의 인접한 두 풍선을 고른 뒤, 두 풍선 중 하나를 터트립니다.
2. 터진 풍선으로 인해 풍선들 사이에 빈 공간이 생겼다면, 빈 공간이 없도록 풍선들을 중앙으로 밀착시킵니다.

여기서 조건이 있습니다. 인접한 두 풍선 중에서 번호가 더 작은 풍선을 터트리는 행위는 최대 1번만 할 수 있습니다. 즉, 어떤 시점에서 인접한 두 풍선 중 번호가 더 작은 풍선을 터트렸다면, 그 이후에는 인접한 두 풍선을 고른 뒤 번호가 더 큰 풍선만을 터트릴 수 있습니다.

당신은 어떤 풍선이 최후까지 남을 수 있는지 알아보고 싶습니다. 위에 서술된 조건대로 풍선을 터트리다 보면, 어떤 풍선은 최후까지 남을 수도 있지만, 어떤 풍선은 무슨 수를 쓰더라도 마지막까지 남기는 것이 불가능할 수도 있습니다.

일렬로 나열된 풍선들의 번호가 담긴 배열 a가 주어집니다. 위에 서술된 규칙대로 풍선들을 1개만 남을 때까지 터트렸을 때 최후까지 남기는 것이 가능한 풍선들의 개수를 return 하도록 solution 함수를 완성해주세요.

 

제한사항

• a의 길이는 1 이상 1,000,000 이하입니다.
  • a[i]는 i+1 번째 풍선에 써진 숫자를 의미합니다.
  • a의 모든 수는 -1,000,000,000 이상 1,000,000,000 이하인 정수입니다.
  • a의 모든 수는 서로 다릅니다.

 

풀이

입출력 예시

 

풀이

가장 작은 수의 풍선과 그 다음으로 작은 풍선은 무조건 살아남을 수 있다는 점은 바로 알 수 있다.

 

다른 풍선들은 어떻게 판별하느냐면...

한 풍선의 위치를 i라고 한다면 0 ~ i - 1까지 있는 풍선들과 i + 1 ~ end 까지의 풍선들이 모두 i보다 크거나, 한쪽만은 더 작아도 된다. 하지만 이를 하나하나 다 비교할 수는 없는 노릇이다.

 

그래서, 왼쪽의 최솟값과 오른쪽의 최솟값을 메모이제이션해준뒤에 비교해준다. 왜 두 곳의 최솟값이냐 하믄, 0 ~ i - 1 (또는 i + 1 ~ end)의 최솟값이 i 위치의 값보다 커야 위 조건을 만족시킬 수 있기 때문이다. ( 결국엔 또 dp구나.. )

 

#include <string>
#include <vector>

using namespace std;

int left_min[1000000];
int right_min[1000000];

int solution(vector<int> a) {
    int answer = 2;
    int a_size = a.size();
    
    // 왼쪽 최솟값
    left_min[0] = a[0];
    for (int i = 1; i < a_size; i++) {
        left_min[i] = min(left_min[i - 1], a[i]);
    }
    
    // 오른쪽
    right_min[a_size - 1] = a[a_size - 1];
    for (int i = a_size - 2; i >= 0; i--) {
        right_min[i] = min(right_min[i + 1], a[i]);
    }
    
    for (int i = 1; i < a_size - 1; i++) {
        if (a[i] < left_min[i - 1] || a[i] < right_min[i + 1]) {
            answer++;
        }
    }
    
    return answer;
}