프로그래머스 / 스택,큐 / 탑

https://programmers.co.kr/learn/courses/30/lessons/42588?language=cpp

코딩테스트 연습 - 탑

#include <string>
#include <vector>
#include <stack>
 
using namespace std;
 
vector<int> solution(vector<int> heights) {
    vector<int> answer;
    stack<int> receivedTower;
    for(int i = heights.size()-1; i >= 0; i--)
    {
        //현재 인덱스에서 스택에 값이 넣어졌는지 판별
        bool isPushed = false;
        //다음 탑의 인덱스부터 검사
        for (int j = i - 1; j >= 0; j--)
        {
            //현재 탑에 인덱스보다 높은 탑을 찾을 경우,(수신할 수 있는 경우)
            if (heights[i] < heights[j])
            {
                //수신할 탑의 인덱스를 스택에 쌓고 , bool변수를 true로 한다.
                receivedTower.push(j + 1);
                isPushed = true;
                break;
            }
        }
        //for문이 돈 후 , 스택에 값이 넣어지지 않았다면 수신이 안된 것.
        //0을 스택에 넣어준다.
        if (isPushed == false)receivedTower.push(0);
    }
    for (int i = 0; receivedTower.size(); i++)
    {
        //반환형이 vector이므로,
        //스택의 끝부터 차례대로 벡터에 넣어준다.
        answer.push_back(receivedTower.top());
        receivedTower.pop();
    }
    return answer;
}

스택을 굳이 안써도 됐지만,

문제 분류가 스택/큐로 되있어서 한번 써보았다.

벡터 answer을 heights.size()로 크기를 할당한 후 ,

수신한 탑의 인덱스를 푸쉬 백 해도 같은 답이 나온다.