[백준/C++] 10773 제로

10773 제로

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

int main()
{
	ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
	vector<int> a;
	int num_case,sum = 0;
	cin >> num_case;
	
	for (int i = 0; i < num_case; ++i)
	{
		int tmp;
		cin >> tmp;
		if (tmp == 0 && a.size() > 0)
			a.pop_back();
		else if (tmp != 0)
		{
			a.push_back(tmp);
		}
	}
	for (auto i : a)
		sum += i;

	cout << sum;
	return 0;
}

• VECTOR의 POP, PUSH 함수를 이용해서 문제를 해결했다.
• 아무것도 없을때 0이 입력으로 들어온다고 뺄 수 없으니 예외처리 하나를 해주었다.