[백준] 10773 제로 - Data Structure / Java

• 문제 링크

10773번: 제로

 

• 풀이 과정

k 개의 정수를 입력받으며 0을 입력받을 경우 pop 하여 이전에 입력받은 값, Stack 의 맨 상단의 값을 제거한다.

문제의 조건 상 이전에 입력받은 값이 있어야 0을 입력받으므로 Stack 이 비어있는 지는 확인하지 않는다.

그외의 입력받은 각각의 정수들은 Stack 에 push 한다.

 

이러한 과정을 거친 후 Stack 에 남아있는 값을 모두 꺼내어 sum 에 더해 정답으로써 출력한다.

 

• 풀이 코드

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

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int k = Integer.parseInt(br.readLine());

        Stack<Integer> stk = new Stack<>();
        for (int i = 0; i < k; i++) {
            int n = Integer.parseInt(br.readLine());
            if (n == 0)
                stk.pop();
            else
                stk.push(n);
        }

        int sum = 0;
        while (!stk.empty())
            sum += stk.pop();

        bw.write(String.valueOf(sum));
        bw.flush();
    }

}