본문 바로가기

Archived(CSE Programming)/알고리즘(C++)

프로그래머스_탑

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

 

코딩테스트 연습 - 탑 | 프로그래머스

수평 직선에 탑 N대를 세웠습니다. 모든 탑의 꼭대기에는 신호를 송/수신하는 장치를 설치했습니다. 발사한 신호는 신호를 보낸 탑보다 높은 탑에서만 수신합니다. 또한, 한 번 수신된 신호는 다른 탑으로 송신되지 않습니다. 예를 들어 높이가 6, 9, 5, 7, 4인 다섯 탑이 왼쪽으로 동시에 레이저 신호를 발사합니다. 그러면, 탑은 다음과 같이 신호를 주고받습니다. 높이가 4인 다섯 번째 탑에서 발사한 신호는 높이가 7인 네 번째 탑이 수신하고, 높이가 7

programmers.co.kr

스택/큐, LV2

기본적인 벡터 문제이다.

스택/큐를 활용하라고 하였는데 벡터로도 구현이 가능하여 다음과 같이 구현하였다.

가장 오른쪽 탑(i)에서부터 왼쪽 탑들(j) 중에서 높이가 높은 탑을 발견하면 그것을 기록하면 된다. 

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

vector<int> solution(vector<int> heights) {
    vector<int> answer(heights.size(),0);
    // 거꾸로 진행
    for(int i=heights.size()-1 ; i>=0; i--){
        int j = 0, found = 0;
        // 왼쪽에서 큰 탑 발견하면
        for(j=i; !found && j>=0; j--)
            if(heights[i] < heights[j])
                found=1;
        // 발견시
        // 마지막 for문에서 -- 하나, 인덱스 0~인것 고려
        if(found==1)
            answer[i]=j+2; 
    }
    return answer;
}