[백준/C++] 17298번_오큰수

이수진·2022년 1월 8일
0
post-custom-banner

문제 링크: https://www.acmicpc.net/problem/17298

문제는 다음과 같습니다.

올해 첫 백준입니다.. 그리고 이 문제는 사실 12월에 풀다가 계속 틀렸습니다가 떠서 두고두고 남겨뒀었던 문제입니다. 며칠만에 다시 푸네요
백준 .. 진짜 오늘부터 무조건 하루에 한문제 이상씩 열심히 풀어야겠습니다! 😊

보이시나요.. 오기가 여기까지 느껴지네용 ㅋㅋ

저는 스택을 이용해서 풀었구요! 제 풀이는 다음과 같습니다.

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

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    vector<int> v, res; // res: 결과벡터
    stack<int> s;
    int n;
    cin>>n;

    for(int i=0; i<n; i++){
        int tmp;
        cin>>tmp;
        v.push_back(tmp);
    }
    reverse(v.begin(), v.end());

    res.push_back(-1);
    s.push(v[0]);

    for(int i=1; i<v.size(); i++){
        if(v[i] < s.top()){
            res.push_back(s.top());
            s.push(v[i]);
        }
        else{ // v[i] >= s.top()

            while(!s.empty() && v[i]>=s.top()){
                s.pop();
            }

            if(s.empty()){
                res.push_back(-1);
                s.push(v[i]);
            }
            else{ // v[i] < s.top()
                res.push_back(s.top());
                s.push(v[i]);
            }

        }
    }

    reverse(res.begin(), res.end());
    for(int i=0; i<res.size(); i++) cout<<res[i]<<" ";
    return 0;
}

for문이 가장 중요한 부분이구요
그동안 왜 계속 틀렸었냐면 바로 같은 수들에 대한 예외처리가 제대로 안되어 있었더라구요

그동안 알고리즘 문제들을 풀며 고민이 많았습니다.
많은 문제들을 풀고 맞는게 중요한가.. 한 문제를 풀고 이렇게 푼 과정이나 로직을 정리하는게 맞는가..
아무래도 저는 후자가 더 맞는 것 같습니다 ㅎㅎ

다른풀이 하나도 가져왔습니다!
참고하세용
로직은 같고, 반복되는 부분이나 겹치는 부분을 다 제거한 풀이입니다

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <math.h>
#include <stack>
using namespace std;
stack<int> st;
int arr[1000001];
vector<int> ans;
int main() {
    ios_base::sync_with_stdio(false), cin.tie(NULL);
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> arr[i];
	}
	for (int i = n; i > 0; i--) {
		while (!st.empty() && st.top() <= arr[i]) {
			st.pop();
		}
		if (st.empty()) ans.push_back(-1);
		else ans.push_back(st.top());

		st.push(arr[i]);
	}
	for (int i = ans.size() - 1; i >= 0; i--) {
		cout << ans[i] << " ";
	}
}
profile
꾸준히, 열심히, 그리고 잘하자
post-custom-banner

0개의 댓글