본문 바로가기

난 이 분야 전문가야!/Algorithm

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

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

 

코딩테스트 연습 - 탑

수평 직선에 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다

programmers.co.kr

 

<풀이>

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#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;
}
cs

 

발상을 거꾸로 해보자.

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

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

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

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

728x90