[백준] 10773. 제로

고재욱·2021년 8월 15일

Baekjoon

목록 보기
2/35

10773. 제로

스택을 활용한 간단한 구현 문제였다.

  1. K개의 숫자를 반복문을 활용해 한개씩 받는다.

    • 입력이 0이면 스택의 top을 pop한다.
    • 입력이 0이 아니면 스택에 넣는다.
  2. 스택에 있는 수를 모두 더한다.

#include <iostream>
#include <stack>
using namespace std;
int main() {
	int k;
	cin >> k;
	stack<int> st;
	for (int i = 0; i < k; i++) {
		int tmp;
		cin >> tmp;
		if (tmp == 0)
			st.pop();
		else {
			st.push(tmp);
		}
	}
	int sum = 0;
	while (!st.empty()) {
		sum += st.top();
		st.pop();
	}
	cout << sum;
}

0개의 댓글