[C++] 10773번 : 제로

https://www.acmicpc.net/problem/10773

10773번: 제로

풀이

스택을 사용하여 정말 쉽게 풀 수 있는 문제입니다.

문제의 '가장 최근에 재민이가 쓴 수를 지우게 시킨다.' 문장에서 이를 가장 효율적으로 수행할 수 있는 자료구조로

스택을 생각할 수 있었다면 구현은 어렵지 않습니다.

 

내 코드

#include <bits/stdc++.h>
using namespace std;

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	int K, sum = 0;
	cin >> K;
	stack<int> s;
	while (K--) {
		int num;
		cin >> num;
		if (num == 0) {
			sum -= s.top();
			s.pop();
		}
		else {
			s.push(num);
			sum += num;
		}
	}
	cout << sum;
	
}

합을 입력받으면서 처리해주지 않아도, while(!s.empty()) 조건을 이용하면, 쉽게 합을 계산할 수 있습니다!