본문 바로가기
Algorithm Study/BaekJoon (JAVA)

백준 10773_제로_JAVA

by 창브로 2024. 4. 23.
728x90

https://www.acmicpc.net/problem/10773

 

10773번: 제로

첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경

www.acmicpc.net

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

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Stack<Integer> stack = new Stack<>();

        int answer = 0;
        int N = Integer.parseInt(br.readLine());

        for (int i = 0; i < N; i++) {
            int num = Integer.parseInt(br.readLine());
            if (num == 0) {
                answer -= stack.pop();
            } else {
                stack.push(num);
                answer += num;
            }
        }

        System.out.print(answer);
    }
}

 

회고

- 스택의 원리는 간단한 문제들을 통해 이해한 것 같다

- 이제 조금 더 응용된 스택 문제를 풀어봐야겠다