[백준,c++] 1874번 - 스택 수열

문제

1874번: 스택 수열

 

코드

#include <iostream>
#include <stack>
#include <vector>

using namespace std;

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    
    int N; cin>>N;
    stack<int>st;
    vector<char> ans;
    
    int curr = 0;
    while(N--){
        int num; cin>>num;
        
        if(st.empty()){
            if(curr>num){
                cout<<"NO";
                return 0;
            }
            int first = curr;
            for(int i=0; i<abs(num-first); i++){
                ans.push_back('+');
                st.push(++curr);
            }
            ans.push_back('-');
            st.pop();
        }
        
        else if(st.top()==num){
            ans.push_back('-');
            st.pop();
        }
        
        else if(st.top()>num){
            while(true){
                if(st.top()==num) break;
                ans.push_back('-');
                st.pop();
            }
        }
        else if(st.top()<num){
            if(curr>num){
                cout<<"NO";
                return 0;
            }
            while(true){
                if(st.top()==num){
                    ans.push_back('-');
                    st.pop();
                    break;
                }
                ans.push_back('+');
                st.push(++curr);
            }
        }
    }
    for(char c: ans) cout<<c<<'\n';
}

 

풀이

코드 구현 자체는 어렵지 않으나 이 문제를 처음 보았을 때 문제 자체가 이해가 가지를 않았습니다. 입출력을 보고 대충 예상은 해서 풀어보았지만 무언가 잘못되었는지 채점했을 때는 오답이 떴고 그렇게 미쳐가던 도중 질문 게시판에서 아래 문제 설명 영상을 찾아서 보게 되었습니다.

위 영상이 없었으면 아직도 문제 이해조차 못했을 것입니다.(제작자 분께 감사드립니다.)  문제 이해가 가지 않으신 분들은 한번 시청하시면 이해가 바로 될 것 같습니다.