[Python][C++] 10773번 제로

문제

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

 

 

코드

1. Python

import sys

input = sys.stdin.readline

k = int(input())

arr = []

for _ in range(k):
    n = int(input())
    if n:
        arr.append(n)
    else:
        arr.pop()

print(sum(arr))

 

 

2. C++

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

int main() {
    int k;

    cin >> k;

    stack<int> s;

    int temp;
    for (int i = 0; i < k; i++) {
        cin >> temp;

        if (temp == 0)
            s.pop();
        else
            s.push(temp);
    }

    int sum = 0;
    while (!s.empty()) {
        sum += s.top();
        s.pop();
    }

    cout << sum;

    return 0;
}

 

 

풀이

이 문제는 스택을 사용하여 간단하게 해결할 수 있다. 입력된 숫자가 0일 경우 스택에서 팝하고, 0이 아닐 경우 스택에 푸시한다. 마지막에 스택에 남아 있는 모든 수의 합을 구하면 된다.