[Baekjoon Online Judge] 10773번: 제로

10773번: 제로

 값을 입력 받으며 0이 나올 때 마다 최근 수를 지우는 문제이다. 현재 돈이 0일 때 LinkedList의 pollLast 메소드를 활용하여 마지막 값을 삭제한다. LinkedList를 활용한 Stack 개념을 활용하여 풀이하였다.

 

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

public class Baekjoon10773 {

    public static void main(String[] args) throws IOException {

        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bufferedWriter = new BufferedWriter(new OutputStreamWriter(System.out));

        int n = Integer.parseInt(bufferedReader.readLine());

        int result = 0;
        LinkedList<Integer> linkedList = new LinkedList<>();
        for (int i = 0; i < n; i++) {
            int currentMoney = Integer.parseInt(bufferedReader.readLine());

            if (currentMoney == 0) {
                linkedList.pollLast();
            } else {
                linkedList.add(currentMoney);
            }
        }

        for (Integer value : linkedList) {
            result += value;
        }

        bufferedWriter.write(String.valueOf(result));

        bufferedWriter.flush();
        bufferedReader.close();
        bufferedWriter.close();
    }
}