algorithm/Stack, Queue, Deque
[C++] BOJ 10773 Zero That Out
gartenhh
2021. 5. 21. 20:13
문제
백준 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에서 빼는 부분을 누락했다....ㅎ