[C++] 백준 10773번 제로

스택 문제입니다.

0을 누르면 pop하고
그 외에는 push를 해줍니다.

스택에 있는 수들을 전부 더해서 출력합니다.

 

#include <iostream>
#include <cstring>
#include <stack>
#include <string>
#include <algorithm>
using namespace std;

int main(int argc, char *argv[])
{
	int n;
	int num;
	int sum = 0;
	stack<int> s;

	cin >> n;

	for (int i = 0; i < n; ++i) {
		cin >> num;
		if (num == 0) {
			s.pop();
		}
		else {
			s.push(num);
		}
	}

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

	cout << sum << endl;

	return 0;
}