본문 바로가기

algorithm/Stack, Queue, Deque

[C++] BOJ 10773 Zero That Out

문제

백준 10773 제로

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

 

10773번: 제로

첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경

www.acmicpc.net

코드

#include<iostream>
#include<stack>
using namespace std;
int k, num, sum;

int main() {
	cin >> k;
	stack<int>s;
	while (k--) {
		cin >> num;
		if (num == 0) {
			sum -= s.top();
			s.pop();
		}
		else {
			sum += num;
			s.push(num);
		}
	}
	cout << sum;
}

풀이방법

숫자를 부를 때마다 더하며 stack에 저장해둔다. 0을 부르면 stack의 top을 sum에서 빼고, pop해준다.

고민과정

대충 읽고 구현하려다가 sum에서 빼는 부분을 누락했다....ㅎ

'algorithm > Stack, Queue, Deque' 카테고리의 다른 글

[C++] BOJ 2346 풍선 터뜨리기  (0) 2021.07.09
[C++] BOJ 1918 후위 표기식  (0) 2021.05.21
[C++] BOJ 1158 요세푸스 문제  (0) 2021.05.21
[C++] BOJ 2161 카드1  (0) 2021.05.21
[C++] BOJ 10828 스택  (0) 2021.05.21