[C++] 백준 10773. 제로

문제

풀이

입력이 0이면 top 값을 sum에서 빼주고 pop한다.
0이 아니면 push하고 sum에 더해준다.
k번 반복한다.

코드

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

int main() {
	stack<int> s;
	int k, tmp;

	cin >> k;
	int sum = 0;

	for (int i = 0; i < k; i++) {
		cin >> tmp;
		if (tmp == 0) {
			sum -= s.top();
			s.pop();
		}
		else {
			s.push(tmp);
			sum += tmp;
		}
	}
	cout << sum;

	return 0;
}

채점