본문 바로가기

코딩 테스트

백준 10773

vector 의 pop, push 연산을 이용하여 풀었다.

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;


int main(void)
{
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	vector<int> nums;

	int input;
	cin >> input;

	for (int i = 0; i < input; ++i)
	{
		int num;
		cin >> num;
		
		if (num == 0)
		{
			if (nums.size() == 0)
			{
				break;
			}
			else
			{
				nums.pop_back();
			}
		}
		else
		{
			nums.push_back(num);
		}
	}

	int out = 0;
	for (int i = 0; i < nums.size(); ++i)
	{
		out += nums[i];
	}
	cout << out;

	return 0;
}

'코딩 테스트' 카테고리의 다른 글

백준 1225  (0) 2023.07.08
soveld 30일 연속 문제 해결  (0) 2023.07.06
백준 7568  (0) 2023.07.05
백준 11866  (0) 2023.07.04
백준 1547  (0) 2023.07.04