백준 18258번 - 큐 2 / Go

문제: https://www.acmicpc.net/problem/18258

18258번: 큐 2

문제 설명

 

문제

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

명령은 총 여섯 가지이다.

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

입력

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

출력

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

예제 입력 1 복사

15
push 1
push 2
front
back
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
front

예제 출력 1 복사

1
2
2
0
1
2
-1
0
1
-1
0
3

---

정답

package main

import (
	"bufio"
	"os"
	"strconv"
	"strings"
)

var (
	rd = bufio.NewReader(os.Stdin)
	wr = bufio.NewWriter(os.Stdout)
)

func main() {
	defer wr.Flush()

	text, _ := rd.ReadString('\n') // 여기서 text는 마지막에 줄바꿈 문자를 포함하므로
	text = strings.TrimSpace(text) // 줄바꿈 문자를 제거해야 함
	n, _ := strconv.Atoi(text)

	queue := newQueue()
	for ; n > 0; n-- {
		str, _ := rd.ReadString('\n') // 여기서 str는 마지막에 줄바꿈 문자를 포함하므로
		str = strings.TrimSpace(str)  // 줄바꿈 문자를 제거해야 함
		arr := strings.Split(str, " ")
		command := arr[0]

		switch command {
		case "push":
			num, _ := strconv.Atoi(arr[1])
			queue.push(num)
		case "pop":
			wr.WriteString(strconv.Itoa(queue.pop()) + "\n")
		case "size":
			wr.WriteString(strconv.Itoa(queue.size()) + "\n")
		case "empty":
			wr.WriteString(strconv.Itoa(queue.empty()) + "\n")
		case "front":
			wr.WriteString(strconv.Itoa(queue.front()) + "\n")
		case "back":
			wr.WriteString(strconv.Itoa(queue.back()) + "\n")
		}
	}

}

type queue struct {
	queue []int
}

func newQueue() queue {
	return queue{[]int{}}
}

func (q *queue) push(x int) {
	q.queue = append(q.queue, x)
}

func (q *queue) pop() (n int) {
	if len(q.queue) == 0 {
		return -1
	}
	n = q.queue[0]
	q.queue = q.queue[1:]
	return
}

func (q *queue) size() int {
	return len(q.queue)
}

func (q *queue) empty() (n int) {
	if len(q.queue) == 0 {
		return 1
	}
	return 0
}

func (q *queue) front() int {
	if q.empty() == 0 {
		return q.queue[0]
	}
	return -1
}

func (q *queue) back() int {
	if q.empty() == 0 {
		return q.queue[q.size()-1]
	}
	return -1
}

 

---

풀이

 프로그래밍 언어로 Queue를 직접 구현할 수 있는지 없는지 파악하는 문제였습니다.

 기존에는 Java, Python으로도 모두 도전해서 구현하고 통과했는데, Go로 하니 시간 초과가 나서 당황한 문제입니다.

 

 해당 문제를 통과하기 위해서는 2가지를 해결해야 합니다.

1. 입력 속도 최적화
2. 출력 속도 최적화

Golang에서는 입력 시 bufio.NewReader(os.Stdin)을 이용하면 입력 문자열 버퍼에 제한 없이 받아들일 수 있다고 하며, 출력 시에는 bufio.NewWriter(os.Stdout)을 이용해 출력해야 해당 문제를 통과할 수 있습니다.