10773번 제로 - C++

[문제]

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

10773번: 제로

 

 

 

---

[풀이]

STL 스택을 이용하는 문제이다.

입력된 정수가 0일 경우에는 가장 최근에 쓴 수를 지우므로, 스택의 empty()를 이용하여, 값이 들어있는 지 확인후, pop() 함수로 삭제하면 된다.

입력된 정수가 0이 아닌 경우에는 원소를 스택에 집어넣기 위해서 push() 함수를 이용하면 된다.

합계 결산은 empty()를 이용하여 값이 있는지 확인하고 pop() 함수를 통해 계산하면 된다.

 

[코드]

#include <bits/stdc++.h>

using namespace std;

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    
    int k;
    cin >> k;
    
    stack<int> S;
    
    while(k--) {
        int n;
        cin >> n;
        if(n == 0) {
            if(!S.empty()) S.pop();
        }
        else S.push(n);
    }
    int sum = 0;
    while(!S.empty()) {
        sum += S.top();
        S.pop();
    }
    cout << sum;
}