[백준 2024/01/17] 10773번 제로

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

10773번: 제로

 

1. 접근 방법

이번 문제를, stack으로 접근할까 고민을 하다가, 

그냥 List로 구현하는 것이 sum을 구하기가 편리할 것이라는 생각이 들어서

list로 구현하였다.

 

2. 코드

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;

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

        int number = Integer.parseInt(br.readLine());
        List<Integer> myList = new ArrayList<>();

        for (int i=0; i<number;i++){
            int new_num = Integer.parseInt(br.readLine());
            if(new_num==0){
                myList.remove(myList.size()-1);
            }
            else{
                myList.add(new_num);
            }

        }
        System.out.println(myList.stream().mapToInt(Integer::intValue).sum());
        pw.flush();
        br.close();
        pw.close();

    }

}