Little Jay

[C++] 백준 10773번 제로 본문

알고리즘/BOJ

[C++] 백준 10773번 제로

Jay, Lee 2021. 3. 21. 18:25

vector 메소드 중에서 pop_back을 사용하는 문제.

아마 이 메소드를 사용하면 알아서 마지막 원소를 삭제해줘서

간단하게 풀 수 있지 않을까 생각한다.

#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;

int main() {

	int k;
	cin >> k;

	vector<int> list;

	for (int i = 0; i < k; i++) {
		int a;
		cin >> a;

		if (a == 0) {
			list.pop_back();
		}
		else {
			list.push_back(a);
		}
	}

	int sum = 0;

	for (int i = 0; i < list.size(); i++) {
		sum += list[i];
	}

	cout << sum << endl;

	return 0;
}
Comments