알고리즘 풀이/백준

[10773] 백준 : 제로(C++)

데롱디롱 2020. 4. 27. 01:57
728x90

 

* 접근한 방법
- 스택이용
   : 0이 입력되면, pop()
   : 0이 아니면, push()

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

int main() {
	int K;
	cin >> K;

	stack<int> pocket;
	for (int i = 0; i < K; i++)
	{
		int a;
		cin >> a;
		if (a == 0)
			pocket.pop();
		else
			pocket.push(a);
	}

	int sum = 0;
	while (1)
	{
		if (pocket.empty())
			break;
		sum += pocket.top();
		pocket.pop();
	}
	cout << sum;
}