본문 바로가기

프로그래머스 문제풀이/LEVEL 2

[C++] 프로그래머스 문제풀이 LEVEL 2 뒤에 있는 큰 수 찾기

 

https://school.programmers.co.kr/learn/courses/30/lessons/154539

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

 

#include <bits/stdc++.h>
using namespace std;

vector<int> solution(vector<int> numbers) {
    vector<int> answer(numbers.size(), -1);
    std::stack<std::pair<int, int>> st;
    for (int i = 0; i < numbers.size(); ++i){
        while(!st.empty() && st.top().first < numbers[i]){
            auto top = st.top(); st.pop();
            answer[top.second] = numbers[i];
        }
        st.push({numbers[i], i});
    }
    return answer;
}