[C++] 백준 - 1874 : 스택 수열

문제 링크

1874번: 스택 수열

 

풀이 )

  현재 입력된 값과 비교하면서 1부터 stack에 집어넣고, 출력을 위한 vector에 '+'를 넣는다.

  현재 입력된 값과 stack의 top이 일치하면 pop하고, 출력을 위한 vector에 '-'를 넣는다.

 

  이 후 stack이 비어있는 경우, 즉 수열을 만들 수 있는 경우 vector의 원소를 모두 출력하고,

  비어있지 않은 경우 "NO"를 출력한다.

#include <iostream>
#include <vector>

using namespace std;

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

    int n;
    cin >> n;

    vector<char> vc;
    vector<int> vi;
    int vi_input = 1;
    while(n--){
        int seq;
        cin >> seq;
        while(vi_input<=seq){
            vi.push_back(vi_input++);
            vc.push_back('+');
        }
        if(vi.back() == seq){
            vi.pop_back();
            vc.push_back('-');
        }
    }
    if(vi.empty()){
        for(vector<char>::iterator it = vc.begin(); it != vc.end(); ++it)
            cout << *it << '\n';
    }
    else
        cout << "NO" << endl;
    return 0;
}

 

주의 )

  결과 출력 시 줄바꿈에 std::endl 사용 시 시간초과가 난다.