알고리즘/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;
}