[ C++ ] #10773 제로

입력되는 숫자를 차곡차곡 스택에 넣다가 0이 입력되면 스택에서 하나를 빼는 방식으로

코드를 짜면 쉽게 해결이 될 것 같다.

 

마지막에는 스택에 있는 값을 모두 더하면 출력값도 쉽게 구할 수 있을 것 같다. 

 

#include <iostream>
#include <stack>

using namespace std;

int main() {

	int K,a,sum=0;
	cin >> K;

	stack<int> s;

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

	}
	a = s.size();
	for (int i = 0; i < a; i++) {
		sum += s.top();
		s.pop();
	}

	cout << sum;
}

 

생각한대로 코드를 짰더니 바로 성공하였다!