백준 10773 제로 Stack with Stack of Java

 

 

---

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

10773번: 제로

 

---

 

import java.util.*;
import java.io.*;
 
public class Main {
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
 
        int K = Integer.parseInt(br.readLine());
        Stack<Integer> stack = new Stack<>();
 
        for(int i=0; i<K; i++) {
            int temp = Integer.parseInt(br.readLine());
 
            if(temp == 0) {
                stack.pop();
            }
            else {
                stack.push(temp);
            }
 
        }
 
        int sum = 0;
        for(int num : stack) {
            sum += num;
        }
 
        System.out.println(sum);
 
    }
}

 

---

스택의 개념을 알고 있는지 물어보는 문제이다. 

 

먼저 BufferReader 객체를 하나 만들어준다. 그런 다음 br의 매서드인 readLine() 함수를 호출해준다.

그렇게 테스트 케이스의 갯수를 k에 저장해 준다. 그 다음 스택을 하나 만들어두고 

테스트 케이스의 갯수만큼 for문을 돌린다.  그리고 다시 사용자로부터 값을 입력 받아 정수형으로 변환하고 temp 변수에 넣어준다. 만약 사용자가 0을 입력했으면 pop으로 스택에 맨 위 (가장 최근에 들어온 값)을 지워주고 그게 아니면 스택에 푸시(더하는 기능)을 해준다. 그리고 for문을 빠져나와 우리가 출력할 변수를 하나 초기화 해준다. 그리고 stack에 있는 모든 값을 위에서 정의한 sum에 넣어주고 그것을 출력하면 끝이다.