BackJoon Algorithm 제로 10773 (Java)

업데이트:
최대 1 분 소요

BackJoon Algorithm - Java

alt

문제

alt

풀이

  • 만약 입력받은 수가 0이면 전에 꺼를 꺼내야하기 때문에 stack[0] 을 먼저 대입한다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

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

        // given
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Stack<Integer> stack = new Stack<>();
        int K = Integer.parseInt(br.readLine());
        int num = Integer.parseInt(br.readLine());
        int sum = 0;

        // 최근거를 지워야하기때문에 stack[0]은 넣고시작
        stack.push(num);
        // when
        for (int i = 1; i < K; i++) {
            num = Integer.parseInt(br.readLine());
            if (num == 0) {
                stack.pop();
            } else {
                stack.push(num);
            }
        }
        for (int i = 0; i < stack.size(); i++) {
            sum += stack.get(i);
        }
        // then
        System.out.println(sum);
        br.close();
    }
}



댓글남기기