Kiki Devlog

[17298번][골4] 오큰수 본문

Coding Test/백준

[17298번][골4] 오큰수

kimkiki 2022. 3. 16. 02:11
728x90

갑자기 골드 난이도가 나와서 시간제한에 걸려 머리 싸메다가 풀이법을 보고 풀었다.

 

내 코드

#include<iostream>
#include <string>
#include <deque>
#include <stack>

using namespace std;


int main() {
	deque<int> d;
	deque<int> ans;
	stack<int> s;
	int totalNum = 0; 
	int num;

	cin >> totalNum;


	for (int i = 0; i < totalNum; i++) {
		cin>> num;
		d.push_back(num);
	}

	for (int i = totalNum-1; i >=0; i--) {
		
		while (!s.empty() && d[i] >= s.top())
			s.pop();

		if (s.empty())
			ans.push_front(-1);
		else {
			ans.push_front(s.top());
		}

		s.push(d[i]);
	}

	for (int i = 0; i < ans.size(); i++)
		cout << ans[i] << " ";
	
	return 0;
}
Comments