[백준] 10773 제로 - Java

문제

링크

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

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

        Stack<Integer> stack = new Stack<>();
        for (int i = 0; i < K; i++) {
            int input = Integer.parseInt(br.readLine());

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

        int sum = 0;
        for (int value : stack) sum += value;

        System.out.println(sum);
    }
}

---

리뷰

처음으로 Stack을 이용하여 풀어본 문제이다.
0이 입력되면 pop 이외의 수가 입력되면 push를 하도록 구현했다.