백준 10845번: 큐

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

10845번: 큐

정수를 저장하고 6개의 명령을 수행하는 큐를 구현하는 문제이다. 구현해야하는 명령은 다음과 같다.

1. push X: X를 큐에 저장한다.

2. pop: 가장 앞에 있는 정수를 삭제하고 출력한다. 만약 큐가 비었다면 -1을 출력한다.

3. size: 큐에 들어 있는 정수의 개수를 출력한다.

4. empty: 큐가 비어있다면 1 아니라면 0을 출력한다.

5. front: 가장 앞에 있는 정수를 출력한다. 만약 큐가 비어있다면 -1을 출력한다.

6. back: 가장 뒤에 있는 정수를 출력한다. 만약 큐가 비어있다면 -1을 출력한다.

 

일단, 기본적인 토대는 저번에 푼 스택 문제에서 사용한 코드에서 가져왔다.

https://pseudo-note.tistory.com/23

백준 10828번: 스택

세부적인 사항을 큐에 맞게 바꾸고 나니 back을 제외한 다른 코드가 완성되었다.

우선, 큐 자체에 back처럼 맨 뒤에 요소를 반환하는 메소드가 있나 찾아봤는데 찾지 못했고,

다음으로 큐가 LinkedList로 구현되니까 LinkedList에 get메소드를 사용할 수 있지 않을까 생각했는데

그것도 되지 않았다.

그러고보니 맨 뒤에 저장된 수는 가장 마지막에 push로 입력된 수 일것이고, 

그렇다면 push에서 입력받은 수를 저장해주는 변수를 만들어주면 간단하게 back을 구현할 수 있음을 깨달았다.

아래는 그렇게 하여 완성된 코드이다.

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;
import java.util.Queue;
import java.util.LinkedList;

public 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 N = Integer.parseInt(br.readLine());
    Queue<Integer> queue = new LinkedList<>();
    int bf = -1;

    for (int i = 0; i < N; i++) {
      StringTokenizer st = new StringTokenizer(br.readLine());
      switch (st.nextToken()) {
        case "push": 
          bf = Integer.parseInt(st.nextToken());
          queue.offer(bf);
          break;
        case "pop": 
          if ( !queue.isEmpty() ) bw.write(queue.remove() + "\n");
          else bw.write("-1\n");
          break;
        case "front": 
          if ( !queue.isEmpty() ) bw.write(queue.peek() + "\n");
          else bw.write("-1\n");
          break;
        case "back":
            if ( !queue.isEmpty() ) bw.write(bf + "\n");
            else bw.write("-1\n");
            break;
        case "empty": 
          if (queue.isEmpty()) bw.write("1\n");
          else bw.write("0\n");
          break;
        default: //size
         bw.write(queue.size() + "\n");
         break;
      }
    }
    br.close();
    bw.flush();
    bw.close();
  }
}