[BaekJoon] 10773번 - 제로 [Java][C++]

문제 링크

---

1. 아이디어

K개의 수에 대해 0이면 최근에 쓴 수를 지우고 아니면 쓴다는 점에서 스택 자료구조를 활용하면 간단하게 해결할 수 있다.

---

2. 코드

1. 풀이 [Java]

import java.io.*;
import java.util.*;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        Deque<Integer> stack = new ArrayDeque<>();

        int k = Integer.parseInt(br.readLine());
        while (k-- > 0) {
            int x = Integer.parseInt(br.readLine());

            if (x == 0) {
                stack.pop();
            } else {
                stack.push(x);
            }
        }

        int sum = 0;
        while (!stack.isEmpty()) {
            sum += stack.pop();
        }

        System.out.println(sum);
    }
}

2. 풀이 [C++]

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

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

    stack<int> st;

    int k;
    cin >> k;

    while (k--) {
        int x;
        cin >> x;

        if (x == 0) {
            st.pop();
        } else {
            st.push(x);
        }
    }

    int sum = 0;
    while (!st.empty()) {
        sum += st.top();
        st.pop();
    }

    cout << sum;
}

---

3. 디버깅

없음.

---

4. 참고

없음.

---