(C++) - 백준(BOJ) 10773 : 제로

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

10773번: 제로

 

간단한 stack 사용 문제였습니다.

 

풀이방법

 1. 0이 연속으로 계속 나와 stack이 비어있는 경우에 pop할 필요가 없습니다.

 2. num이면 num을 push() num이 0이면 pop()

 3. stack이 빌 때까지 안의 원소들을 합해줍니다.

Code

#include <bits/stdc++.h>
using namespace std;
int main(){
    stack <int> s;
    int k,ans=0;
    cin >> k;
    for(int i = 0; i < k; i++){
        int num;
        cin >> num;
        if(num) s.push(num);
        else s.pop();
    }
    while(!s.empty()){ 
        ans += s.top(); 
        s.pop(); 
    }
    cout << ans <<'\n';
}