[BaekJoon] 17298번 : 오큰수

• 이 글은 c++로 풀이를 작성하였습니다.

문제

 

예시

 

풀이

• 처음 문제를 보면 떠오르는 풀이는 O(n2)방법으로 2중 for문으로 전체 탐색하면서 비교하는 방법입니다.
• 하지만, 이 문제는 경우의 수가 1,000,000이기 때문에 당연히 시간초과가 날 것이라고 생각해야합니다.
• stack을 활용하면 O(n)으로 풀이가 가능합니다.

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

int main() {
	// your code goes here
	int num;
	cin>>num;
	
	vector<int> vec;
	for(int i=0;i<num;i++){
		int n;
		cin>>n;
		vec.push_back(n);
	}
	
	stack<int> s;
	for(int i=vec.size()-1;i>0;i--){
		s.push(vec[i]);
	}
	
	int data = vec[0];
	int index = 0;
	int cnt = 0;
	while(!s.empty()){
		if(data<s.top()){
			for(int i=0;i<=cnt;i++){
				cout<<s.top()<<' ';
			}
			s.pop();
			data = vec[++index];
			cnt = 0;
		}else{
			s.pop();
			cnt++;
		}
	}
	cout<<-1;
	
	return 0;
}

• 추가로 궁금한 점이나 수정할 부분 있으면 댓글로 남겨주세요.