[백준] 10773 제로

문제 설명

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

10773번: 제로

문제 풀이법

일단 k개의 수를 입력 받은 후 stack을 선언하여 입력 받은 수가 0이 아닐때는 stack에 push 해준다

문제에서 입력받은 수가 0일 경우 지울 수가 있음을 보장할 수 있다고 하였으므로, 입력 받은 수가 0이면 pop을 해준다 

 

소스 코드

#include <iostream>
#include <stack>

using namespace std;

int main()
{
    int n;
    
    cin >> n;
    
    stack<int> s;
    for (int i = 0; i < n; i++)
    {
        int tmp;
        cin >> tmp;
        
        if (tmp != 0)
            s.push(tmp);
        else
            s.pop();
    }
    
    int ans = 0;
    while (!s.empty())
    {
        ans += s.top();
        s.pop();
    }
    printf("%d\n", ans);
}