[백준] 10773번 제로

스택 자료구조를 이용하면 편하게 풀 수 있는 문제였습니다.

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

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

	return 0;
}