17299 오등큰수 with C++

문제

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

17299번: 오등큰수

 

해결한 방법

이 전 문제인 17298의 개선한 방법을 적용한 방법으로 해결했습니다.

https://everydaystudy.tistory.com/47

 

#include <iostream>
#include <stack>
#define MAX 1000001
using namespace std;

int count_li[MAX];
int input_li[MAX];
int result[MAX];

int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);

	int n;
	cin >> n;

	for (int i = 0; i < n; i++){
		cin >> input_li[i];
		count_li[input_li[i]]++;
	}

	stack<int> num_stack;
	for (int i = n-1; i >= 0; i--){
		while (!num_stack.empty() && count_li[num_stack.top()] <= count_li[input_li[i]])
			num_stack.pop();
		
		if (!num_stack.empty() && num_stack.top() != input_li[i])
			result[i] = num_stack.top();
		else
			result[i] = -1;

		num_stack.push(input_li[i]);
	}

	for (int i = 0; i < n; i++){
		cout << result[i] << " ";
	}

	return 0;
}