포스트

[ 프로그래머스 ] 뒤에 있는 큰 수 찾기

문제


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

풀이


배열 의 각 원소들에 대해 자신보다 뒤에 있는 숫자 중에서 자신보다 크면서 가장 가까이 있는 수를 찾는 문제로, 순서대로 값을 비교하여 뒷 큰수를 찾는다.

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
31
32
33
#include <string>
#include <vector>
#include <stack>

using namespace std;

vector<int> solution(vector<int> numbers) 
{
    vector<int> answer(numbers.size(), -1);

    stack<pair<int,int>> st;
    int idx = 0;

    for (int i = 0; i < numbers.size(); i++)
    {
        while (!st.empty())
        {
            if (st.top().first >= numbers[i])
            {
                break;
            }

            answer[st.top().second] = numbers[i];
            st.pop();
        }

        st.push({ numbers[i],i });
    }


    return answer;
}

이 기사는 저작권자의 CC BY 4.0 라이센스를 따릅니다.