백준 10773번 [제로](C++)-yes6686- 티스토리

백준 문제 풀이: 10773 [제로]

---

문제 링크: https://www.acmicpc.net/problem/10773

문제 설명:

주어진 입력에 따라 숫자를 스택에 추가하거나 제거하는 작업을 수행합니다. 숫자 0이 입력되면, 가장 최근에 추가된 숫자를 제거합니다. 최종적으로 스택에 남아 있는 숫자의 합을 출력하는 문제입니다.

---

문제 해결 코드

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

int main() {
    vector v;
    int n, x;
    cin >> n;
    for (int i = 0; i < n; i++) {
        cin >> x;
        if (x == 0) {
            v.pop_back();
        } else {
            v.push_back(x);
        }
    }
    int sum = 0;
    for (int i = 0; i < v.size(); i++) {
        sum += v[i];
    }
    cout << sum;
}

---

코드 설명

• 핵심 알고리즘: 스택을 사용하여 숫자를 추가하거나 제거
• 구현 세부사항:
  • vector를 스택처럼 사용하여 push_back과 pop_back를 활용
  • 입력된 숫자가 0이면 스택에서 가장 최근에 추가된 숫자를 제거
  • 모든 입력 처리가 끝난 후, 스택에 남아 있는 숫자의 합을 계산
• 시간 복잡도: O(n)
  • 입력 처리: O(n)
  • 남아 있는 숫자 합 계산: O(n)

---

결과

스택을 사용하여 입력된 숫자와 0에 따른 처리 결과를 구현하였습니다. 최종적으로 스택에 남아 있는 숫자의 합을 출력합니다.

다른 접근 방식이나 개선 사항이 있다면 댓글로 공유 부탁드립니다!