본문 바로가기

알고리즘 문제

[C/C++ 백준 10773번] 제로 (Silver 4)

www.acmicpc.net/problem/10773

 

10773번: 제로

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

www.acmicpc.net

기본적인 스택을 사용하는 문제이다.

 

#include <cstdio>
#include <stack>
using namespace std;
int main(void){
	int K, x, cnt=0;
	stack <int> stk;
	scanf("%d", &K);
	for(int i=0; i<K; i++){
		scanf("%d", &x);
		if(x!=0)
			stk.push(x);
		else
			stk.pop();
	}
	while(!stk.empty()){
		cnt += stk.top();
		stk.pop();
	}
	printf("%d", cnt);
}