[백준(BOJ)] 10845 큐 - 스위프트 (Swift)

[Silver IV] 큐 - 10845

문제 링크

성능 요약

메모리: 69376 KB, 시간: 28 ms

분류

자료 구조, 큐

제출 일자

2023년 12월 31일 16:50:53

문제 설명

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

명령은 총 여섯 가지이다.

• push X: 정수 X를 큐에 넣는 연산이다.
• pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 큐에 들어있는 정수의 개수를 출력한다.
• empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
• front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

 

풀이

이 문제는 전형적인 큐 문제로, 큐에 대한 성질만 이해하고 있으면 풀 수 있습니다.

큐에 대한 설명은 전에도 작성하였으니, 생략하도록 하겠습니다.

 

다음은 코드입니다.

// Queue
struct Queue {
    var enqueueStack = [Int]()
    var dequeueStack = [Int]()


    mutating func push(_ value: Int) {
        enqueueStack.append(value)
    }

    mutating func pop() -> Int {
        // 만약 dequeue에 값이 없다면 enqueue에 있는 값들을 전부 옮김
        if dequeueStack.isEmpty {
            while let value = enqueueStack.popLast() {
                dequeueStack.append(value)
            }
        }

        // dequeue에서 값을 pop
        return dequeueStack.popLast() ?? (enqueueStack.isEmpty ? -1 : enqueueStack.removeFirst())
    }

    func size() -> Int {
        return enqueueStack.count + dequeueStack.count
    }

    func empty() -> Int {
        return size() == 0 ? 1 : 0
    }

    func front() -> Int {
        if let value = dequeueStack.last {
            return value
        } else if let value = enqueueStack.first {
            return value
        } else {
            return -1
        }
    }

    func back() -> Int {
        if let value = enqueueStack.last {
            return value
        } else if let value = dequeueStack.first {
            return value
        } else {
            return -1
        }
    }
}

let n = Int(readLine()!)!
var queue = Queue()

for _ in 0..<n {
    let input = readLine()!.split(separator: " ")
    let command = String(input[0])
    switch command {
    case "push":
        let value = Int(input[1])!
        queue.push(value)
    case "pop":
        print(queue.pop())
    case "size":
        print(queue.size())
    case "empty":
        print(queue.empty())
    case "front":
        print(queue.front())
    case "back":
        print(queue.back())
    default:
        break
    }
}