[BOJ/백준] C++ 10773번 제로 문제 풀이 (벡터 사용)

minjgziii·2022년 7월 30일
0
post-thumbnail

문제

풀이

  • 숫자를 담을 벡터 선언
  • for문을 이용해 벡터에 숫자 push_back
  • 벡터의 원소가 0일 때, 그 원소바로 전 인덱스의 원소 제거
  • accumulate를 사용하여 벡터의 모든 원소의 총합 출력

코드

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

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	int N = 0;
	int num = 0;
	vector<int> v;
	

	cin >> N;

	for (int i{ 0 }; i < N; i++) {
		cin >> num;
		v.push_back(num);
	}

	for (int j{ 0 }; j < v.size(); j++) {
		if (v[j] == 0) {
			v.erase(v.begin() + j);
			v.erase(v.begin() + j - 1);
			j -= 2;
		}
	}

	cout << accumulate(v.begin(), v.end(), 0);
}
profile
티스토리로 이사갑니당

0개의 댓글