[프로그래머스] LEVEL2 탑문제(C++)

프로그래머스라는 사이트의 문제의 해설을 올려보고자 한다. 문제는 아래 URL에 접속해서 읽어보면 될 듯 싶다.

https://programmers.co.kr/learn/courses/30/lessons/42588

#include <string>
#include <vector>
using namespace std;
 
vector<int> solution(vector<int> heights) {
 
    vector<int> answer;
    answer.resize(heights.size(),0);
    int nowheight;
 
    for (int i =heights.size()-1;i>=0;i--) {
        int j= i-1;
        nowheight = heights[i];
        while(heights[j] <= nowheight) {
            j--;
        }
        j = j+1;
        if(j<0) {
            j =0;
        }
        answer[i] =j;
    }
 
    return answer;
}

이 문제를 푼 방식은 간단하다. vector에 저장된 요소들을 하나씩 꺼내서 살펴보면서 그것보다 앞쪽에 있는 요소들과 값을 비교하여 꺼낸 요소보다 큰 값이 발견되면 while을 안들어게되고 그 부분에 값을 업데이트하게 된다. 쉽게 말하면 for문을 두번 돌려서 값을 비교하되 특정조건이 만족되면 break를 걸어준다는 소리이다.

출처 - 프로그래머스(https://programmers.co.kr/)