[JAVA53] 28278. 스택2

28278. 스택2 (S4)

시간 제한	메모리 제한	제출	정답	맞힌 사람	정답 비율
2 초	1024 MB	38963	14205	11829	36.946 %

 

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

1. 1 X: 정수 X를 스택에 넣는다. (1 ≤ X ≤ 100,000)
2. 2: 스택에 정수가 있다면 맨 위의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.
3. 3: 스택에 들어있는 정수의 개수를 출력한다.
4. 4: 스택이 비어있으면 1, 아니면 0을 출력한다.
5. 5: 스택에 정수가 있다면 맨 위의 정수를 출력한다. 없다면 -1을 대신 출력한다.

입력

 

첫째 줄에 명령의 수 N이 주어진다. (1 ≤ N ≤ 1,000,000)

둘째 줄부터 N개 줄에 명령이 하나씩 주어진다.

출력을 요구하는 명령은 하나 이상 주어진다.

출력

출력을 요구하는 명령이 주어질 때마다 명령의 결과를 한 줄에 하나씩 출력한다.

---

summary

명령어에 따라 결과 출력

 

strategy

스택 활용

• 스택에서 값을 제거할 때는 항상 비어있는지 체크 (empty)
• 스택 정수 개수는 따로 계산 필요
• 값 저장 (push)
• 값 읽기 (peek)
• 값 제거 (pop)

 

note

• 1 ≤ N ≤ 1,000,000

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

public class Main {
    static int N;
    static int top = -1;
    static Stack<Integer> stack = new Stack<>();
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder answer = new StringBuilder();

        N = Integer.parseInt(br.readLine()); // 명령 개수

        for(int n = 0; n < N; n++) {
            StringTokenizer input = new StringTokenizer(br.readLine());

            int num = Integer.parseInt(input.nextToken());

            if(num == 1) {
                stack.add(Integer.parseInt(input.nextToken())); // num = 1 데이터 저장
                top++;
            } else if(num == 2) {
                if(stack.isEmpty()) answer.append(-1).append("\n");
                else {
                    answer.append(stack.pop()).append("\n");
                    top--;
                }
            } 
            else if(num == 3) answer.append(top + 1).append("\n");
            else if(num == 4) {
                if(stack.isEmpty()) answer.append(1).append("\n");
                else answer.append(0).append("\n");
            } else if(num == 5) {
                if(stack.isEmpty()) answer.append(-1).append("\n");
                else {
                    answer.append(stack.peek()).append("\n");
                }
            }
        }

        System.out.println(answer);
    }
}

문제	결과	메모리	시간	언어코드	길이
28278	맞았습니다!!	238868 KB	932 ms	Java 11 / 수정	1462 B

 

memo

• stack 은 vector 메소드 사용 가능하 (remove, add, isEmpty, ...)
• 자바에서 stack 은 성능저하를 일으키므로 ArrayDeque, ArrayList 활용 권장

자바의 Stack 클래스는 왜 사용하지 않는 걸까?

• 메모리제한은 왜 이렇게 큰 걸까?