[Java] BOJ 10773 제로

문제 

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

• 첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000)정수가 "0"일 경우에 지울 수 있는 수가 있음을 보장할 수 있다.
• 이후 K개의 줄에 정수가 1개씩 주어진다.
• 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경우 해당 수를 쓴다.
• 재민이가 최종적으로 적어 낸 수의 합을 출력한다.
• 최종적으로 적어낸 수의 합은 231-1보다 작거나 같은 정수이다.

아이디어

• 숫자가 들어올때마다 stack에 담고 만약 숫자가 0일 경우, stack의 맨 위에 있는 숫자를 제거한다.
• 모든 숫자를 입력받고 난 후, While문을 통해 stack이 빌 때까지 stack에 있는 숫자들을 더해준다. 

겪은 시행착오

• X

오늘은 모두 stack 문제라 첫문제를 풀고 남은 문제는 수월하게 풀 수 있었던 것 같다 ㅎㅎ

코드

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

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

        // 입력이 들어올 수
        int N = Integer.parseInt(br.readLine());

        // 수를 저장할 stack
        Stack<Integer> num = new Stack<>();

        int result = 0;

        for(int i = 0; i < N; i++){
            // 한 줄에 숫자를 하나씩 입력받음
            int n = Integer.parseInt(br.readLine());

            // 만약 입력받은 숫자가 0일 경우
            if(n == 0){
                // 맨 위에 있는 숫자 pop
                // 정수가 0일 경우, 지울 수 있는 수가 있음을 보장할 수 있다고 했으므로
                num.pop();
            }
            // 아닐 경우, stack에 숫자를 넣음
            else{
                num.add(n);
            }
        }

        // stack이 비기 전까지 stack에 있는 값을 모두 더함
        while(!num.empty()){
            result = result + num.pop();
        }

        System.out.println(result);
    }
}