10773 - 제로

• 난이도: 실버 4
• 날짜: 2023년 2월 10일
• 상태: Correct
• 추가 검토 여부: No
• 알고리즘 : stack

solution

• stack자료구조를 사용해서 0이 입력되면 pop을 수행하고 그 이외의 경우에 대해 push를 수행한다.

code

#include <iostream>
#include <stack>
using namespace std;

int main()
{
    int N;
    cin >> N;
    int input;
    stack<int> stack;
    for(int i = 0; i < N; i++)
    {
        cin >> input;
        if(!input)
        {
            stack.pop();
        }
        else
        {
            stack.push(input);
        }
    }
    int sum = 0;
    while(!stack.empty())
    {
        sum += stack.top();
        stack.pop();
    }
    cout << sum << endl;

    return 0;
}

ref

10773번: 제로