프로그래머스 :: 탑 - c++ 풀이 공유

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;
    int j;
    answer.push_back(0);
    
    for(int i = 1;i<heights.size();i++){
            if(heights[i] < heights[i - 1]){
                answer.push_back(i);
            }else{
                j = answer.size() - 1;
                while(true){
                    if(heights[answer[j] - 1] == 0){
                        answer.push_back(0);
                        break;
                    }
                    if(heights[answer[j] - 1] > heights[i]){
                        answer.push_back(answer[j]);
                        break;
                    }
                    j--;
                }
            }
    }
    return answer;
}

 

발상을 거꾸로 해보자.

자신의 바로 왼쪽 탑에 수신이 안될경우 바로 왼쪽 탑보다 높은 탑에 수신이 된다.

송신하는 탑의 바로 왼쪽 탑보다 높은 탑의 위치는 왼쪽 탑의 answer 배열에 저장되어 있다.

외쪽 탑보다 높은 탑으로 보다 높으로 answer 배열을 이용하여 점프 하면서 현재 송신 탑보다 높은 탑을 탐색한다.

answer[j]가 0일경우 해당 탑의 신호를 받을 수 있는 탑은 없다는 것을 나타내고 while문을 탈출한다.