[SWEA] 8931번 - 제로 [Java]

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AW5jBWLq7jwDFATQ

 

[SWEA] 8931번 - 제로 [Java]

[SWEA] 8931번 - 제로 [Java]

---

1.  아이디어

 

Stack 자료구조를 활용하면 간단하게 해결할 수 있다.

---

2. 문제풀이

 

0이면 Stack에서 pop, 0이 아니면 Stack에 push를 하면 간단하게 영수증을 구현할 수 있다.

---

3. 코드

 

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

public class Solution {
    public static void main(String[] args) throws IOException {
//        BufferedReader br = new BufferedReader(new InputStreamReader(Solution.class.getResourceAsStream("sample_input.txt")));
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();

        int T = Integer.parseInt(br.readLine());
        for (int tc = 1; tc <= T; tc++) {
            int K = Integer.parseInt(br.readLine());

            Deque<Integer> stack = new ArrayDeque<>();
            for (int i = 0; i < K; i++) {
                int num = Integer.parseInt(br.readLine());

                if (num == 0) stack.pop();
                else stack.push(num);
            }

            int sum = 0;
            while (!stack.isEmpty()) {
                sum += stack.pop();
            }

            sb.append("#").append(tc).append(" ").append(sum).append("\n");
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기