문제풀이/백준

[C++] 백준 - 10773 : 제로

orubt 2020. 12. 1. 00:56
반응형

문제 링크

 

10773번: 제로

첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경

www.acmicpc.net

 

풀이 )

  "0"이 지울 수 있음이 보장되어있으므로, 0이 아니면 push, 0이면 pop한 뒤

  마지막에 stack에 있는 모든 원소들을 더해준다.

 

#include <iostream>
#include <stack>

using namespace std;

int main(void){
    int n;
    stack<int> st;
    cin >> n;
    while(n--){
        int input;
        cin >> input;
        if(input)
            st.push(input);
        else
            st.pop();
    }
    int sum = 0;
    while(!st.empty()){
        sum += st.top();
        st.pop();
    }
    cout << sum << endl;
    return 0;
}
반응형