[BOJ] 17298번 - 오큰수

<접근 방식>

스택을 활용하여 문제를 풀었다.

 

조건)

자기 자신 보다 오른쪽에 있는 수 중 크고 가장 왼쪽에 있는 숫자를 출력하고 없는 경우에는 -1을 출력한다.

 

vector v에 입력값을 넣고 vector ans는 -1 값으로 초기화 시켜준다.

문제의 조건을 만족하고 stack이 비어있지 않을 경우 vector ans에 값을 넣어준다.

<전체 코드>

#include <iostream>
#include <vector>
#include <stack>
using namespace std;

int N;

int main()
{
	cin.tie(0);
	ios_base::sync_with_stdio(false);
	
	cin >> N;
	
	vector<int> v;
	int num;
	for (int i = 0; i < N; i++) {
		cin >> num;
		v.push_back(num);
	}
	
	vector<int> ans(N, -1);
	stack<int> s;

	for (int i = 0; i < N; i++) {
		while (!s.empty() && v[s.top()] < v[i]) {
			ans[s.top()] = v[i];
			s.pop();
		}
		s.push(i);
	}
	for (int i = 0; i < N; i++)
		cout << ans[i] << " ";
}

<문제 링크>

https://www.acmicpc.net/problem/17298

17298번: 오큰수