[Java] 백준(실버-4) 10773번 - 제로

자세한 입출력 테스트케이스는 문제 링크를 참고하자.

문제 풀이

---

이번 문제는 스택의 후입 선출(LIFO) 구조를 활용하면 손쉽게 풀 수 있다.
K개의 주어지는 정수가 0 이상이면 스택에 저장하고, 0이라면 맨 위 값을 추출하면 된다.

아이디어 도출

• 주어지는 정수가 0보다 크면 스택에 삽입한다.(push연산)
• 주어지는 정수가 0이라면 스택에서 추출한다.(pop연산)
• 모든 정수 입력이 끝나면 스택에 남은 값의 합을 구한다.

아이디어를 토대로 코드를 작성해보자.

int K = Integer.parseInt(br.readLine());
Stack<Integer> st = new Stack<>();

먼저 정수의 갯수 K를 입력받고 K개의 0보다 큰 정수를 담을 스택을 하나 생성하자.

for(int i=0; i<K; i++) {
    int num = Integer.parseInt(br.readLine());
    if(num > 0) st.push(num);
    else st.pop();
}

K개의 정수를 입력받으며 0보다 크다면 push연산을 통해 스택에 삽입하고, 정수가 0이라면 pop연산을 통해 스택에서 추출한다.

int sum = 0;
if(!st.empty()) {
    while(!st.empty()) {
        sum += st.pop();
    }
}

정수 입력이 끝나면 스택에 값이 있는지 확인 후 값이 있다면 값들의 합을 구하면 된다.

작성코드

---

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

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> st = new Stack<>();

        for(int i=0; i<K; i++) {
            int num = Integer.parseInt(br.readLine());
            if(num > 0) st.push(num);
            else st.pop();
        }
        
        int sum = 0;
        if(!st.empty()) {
            while(!st.empty()) {
                sum += st.pop();
            }
        }
        
        bw.write(sum+"\n");
        
        bw.flush();
        bw.close();
        br.close();
    }
}

회고

• 스택의 장바구니 속성, 즉 후입선출(LIFO) 기능을 통해 주어지는 정수마다 push연산과 pop연산을 하여 최종 스택에 존재하는 값들의 합을 구할 수 있었다.