[백준 알고리즘] 10773번 : 제로

문제

10773번 : 제로

10773번: 제로

내가 작성한 코드 (C++ 성공)

#include <iostream>
#include <string.h>
#include <vector>

using namespace std;

int main(){
	int k;
	cin >> k;
	
	int money;
	
	vector<int> wallet;
	
	for(int i=0; i<k; i++){
		cin >> money;
		if(money == 0) {
			if(wallet.size() != 0){
				wallet.pop_back();
			}
		} else {
			wallet.push_back(money);
		}
	}
	
	int sum = 0;
	
	for(int i=0; i<wallet.size(); i++){
		sum += wallet[i];
	}
	
	cout << sum;
	
	return 0;
}

 

정리

vector를 이용해서 문제를 풀어보았다.
vector에는 기본적으로 스택에 대한 함수들을 이용할 수 있기 때문에 스택에 대한 문제를 쉽게 풀 수 있었다.

따라서, 0이란 숫자가 들어오게 되면 pop을 해주면서 빼주고 아닌 경우 push를 해주었다.