[C++/스택] BOJ 10773 제로

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

10773번: 제로

#include <iostream>
#include <stack>

using namespace std;

int main()
{
    int k, input, sum;
    stack<int> nums;

    cin >> k;

    while(k--){
        cin >> input;
        if(input == 0) nums.pop();
        else nums.push(input);
    }

    sum = 0;

    int size = nums.size();
    for(int i = 0; i < size; i++){
        sum += nums.top();
        nums.pop();
    }
    cout << sum;
}

마지막 sum에 stack 내 모든 수들의 합을 구하는 for문에서 size를 고정 안시키고 돌려서 돌아갈 때마다 사이즈가 줄어드는 오류를 만드는 실수를 했다.

 

더 깔끔하게는 어떻게 만드나요...ㅠㅠ