[백준/Java] 큐 2 18258

𝙰𝚕𝚐𝚘𝚛𝚒𝚝𝚑𝚖/𝙱𝚊𝚎𝚔𝚓𝚘𝚘𝚗

[백준/Java] 큐 2 18258

해버니 2023. 11. 15. 16:48

문제

풀이

"큐2"는 "큐"라는 문제에서 "큐의 가장 뒤에있는 정수를 출력한다" 라는 기능이 추가된 문제이다.

그렇게 풀었는데 처음에는 시간 초과가 떴다.

System.out. 으로 바로 바로 출력했는데

좀 더 빠른 BufferedWrtier를 이용해서

담아놨다가 뿌려주었더니 시간초과가 해결됐다.

그리고 나서 다시 제출했는데 틀렸습니다라고 떴다.

알고 봤더니 큐의 가장 뒤에있는 정수를 출력하고 없을 경우 -1을 출력하라고 했는데

-1을 출력하는 if문을 만들지 않았었다.

그리고 pop 함수를 만들 때 queue.isEmpty()가 아닌 queue.size() >0 이라 썼는데

queue.isEmpty()로 해야 더 정확하고 문맥상 맞는 메소드 같다.

그렇게해서 제출했더니 "맞습니다"라고 떴다.

1. String[] input = br.readLine().split("\\s");

String 배열로 입력값 받기

package algorithm;

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

public class Main {

    // 정수 x를 큐에 넣기
    public int pushNum(Queue<Integer> queue, int num) {
        queue.offer(num);
        return num;
    }

    // 큐에서 가장 앞에 있는 정수 출력, 없을 경우 -1 출력
    public int getFront(Queue<Integer> queue) {
        if (queue.isEmpty()) {
            return -1;
        } else {
            return queue.peek();
        }
    }

    // 큐의 가장 뒤에 있는 정수 출력, 없을 경우 -1 출력
    public int getBack(Queue<Integer> queue, int back) {
        if (queue.isEmpty()) {
            return -1;
        } else {
            return back;
        }
    }

    // 큐에 들어있는 정수의 개수
    public int getSize(Queue<Integer> queue) {
        return queue.size();
    }

    // 큐가 비어있으면 1, 아니라면 0
    public int getEmpty(Queue<Integer> queue) {
        if (queue.isEmpty()) {
            return 1;
        } else {
            return 0;
        }
    }

    // 큐 삭제하기
    public int popFront(Queue<Integer> queue) {
        if (queue.isEmpty()) {
            return -1;
        } else {
            return queue.poll();
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        Queue<Integer> queue = new LinkedList<>();
        int current = 0;
        Main queueCalc = new Main();

        int n = Integer.parseInt(br.readLine());
        for (int i = 0; i < n; i++) {
            String[] input = br.readLine().split("\\s");

            switch (input[0]) {
                case "push":
                    current = queueCalc.pushNum(queue, Integer.parseInt(input[1]));
                    break;
                case "front":
                    bw.write(queueCalc.getFront(queue) + "\n");
                    break;
                case "back":
                    bw.write(queueCalc.getBack(queue, current) + "\n");
                    break;
                case "size":
                    bw.write(queueCalc.getSize(queue) + "\n");
                    break;
                case "empty":
                    bw.write(queueCalc.getEmpty(queue) + "\n");
                    break;
                case "pop":
                    bw.write(queueCalc.popFront(queue) + "\n");
                    break;
            }
        }
        bw.flush();
        bw.close();

    }
}

2. StringTokenizer st = new StringTokenizer(br.readLine());

String command = st.nextToken();

바로 nextToken()으로 받기

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

public class Main {

    // 정수 x를 큐에 넣기
    public int pushNum(Queue<Integer> queue, int num) {
        queue.offer(num);
        return num;
    }

    // 큐에서 가장 앞에 있는 정수 출력, 없을 경우 -1 출력
    public int getFront(Queue<Integer> queue) {
        if (queue.isEmpty()) {
            return -1;
        } else {
            return queue.peek();
        }
    }

    // 큐의 가장 뒤에 있는 정수 출력, 없을 경우 -1 출력
    public int getBack(Queue<Integer> queue, int back) {
        if (queue.isEmpty()) {
            return -1;
        } else {
            return back;
        }
    }

    // 큐에 들어있는 정수의 개수
    public int getSize(Queue<Integer> queue) {
        return queue.size();
    }

    // 큐가 비어있으면 1, 아니라면 0
    public int getEmpty(Queue<Integer> queue) {
        if (queue.isEmpty()) {
            return 1;
        } else {
            return 0;
        }
    }

    // 큐 삭제하기
    public int popFront(Queue<Integer> queue) {
        if (queue.isEmpty()) {
            return -1;
        } else {
            return queue.poll();
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        Queue<Integer> queue = new LinkedList<>();
        int current = 0;
        Main queueCalc = new Main();

        int n = Integer.parseInt(br.readLine());
        
        for (int i = 0; i < n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            String command = st.nextToken();
            
            switch (command) {
                case "push":
                    current = queueCalc.pushNum(queue, Integer.parseInt(st.nextToken()));
                    break;
                case "front":
                    bw.write(queueCalc.getFront(queue) + "\n");
                    break;
                case "back":
                    bw.write(queueCalc.getBack(queue, current) + "\n");
                    break;
                case "size":
                    bw.write(queueCalc.getSize(queue) + "\n");
                    break;
                case "empty":
                    bw.write(queueCalc.getEmpty(queue) + "\n");
                    break;
                case "pop":
                    bw.write(queueCalc.popFront(queue) + "\n");
                    break;
            }
        }
        bw.flush();
        bw.close();

    }
}

정답

2번이 더 빠르다.