[BOJ 백준] 10773번 제로 / C++

문제

---

문제를 보시려면 링크를 클릭해주세요.

풀이

---

STL stack을 사용하여 풀었습니다.

입력값이 0 or 나머지 숫자 인 경우만 나눠서 간단하게 풀 수 있는 문제입니다.

소스 코드

---

#include <iostream>
#include <stack>


using namespace std;

int main(int argc, char**argv) {

	ios::sync_with_stdio(0);
	cin.tie(0);
		
	stack<int> S;
	int K;
	int sum=0;
	
	cin >> K;
	
	while(K--){
		int N;
		cin >> N;
		
		if(N){
			S.push(N);
			sum+=N;
		}
		
		else{
			if(!S.empty()){
				sum-=S.top();
				S.pop();
			}
		}
	}
	
	cout << sum;
	
	return 0;
}