10-wkdghdwns199

[wkdghdwns199]

🔗 문제 링크

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

✔️ 소요된 시간

1시간 - 왜 시간초과가 뜨는거야..? 라고 보다가 알고보니 O(n) 의 시간 복잡도를 가지는 함수를 사용해서 그렇다..!

✨ 수도 코드

충격과 공포의 시간초과...

Escape 를 못하는 문제였는줄 알았는데, 그렇지 않았고 결론부터 이야기 하면 list 에서의 pop(0) 함수는 O(n) 의 시간 복잡도를 가지기 때문에 시간 초과가 난 것입니다...! (이거 때문에 1시간 걸림..)

다른 문제에서도 굳이 필요한 작업이 아니면 deque 를 써서 popleft() 함수를 쓰는 것이 바람직하겠습니다...!

시간 복잡도에 걸린 코드

import sys
N = int(input())
queue = list()
for _ in range(N):
    _str = sys.stdin.readline().split()
    if 'push' == _str[0] :
        queue.append(_str[1])
    elif 'pop' == _str[0] :
        if len(queue) == 0 : print(-1)

        # 0 번째 요소를 pop 하면 뒤에 있던 요소들을 하나씩 당겨오는 작업 때문에 O(n) 시간이 걸린다.
        else : print(queue.pop(0)) 
    elif 'size' == _str[0] :
        print(len(queue))
    elif 'empty' == _str[0] :
        if len(queue) == 0 : print(0)
        else : print(1)
    elif 'front' == _str[0] :
        if len(queue) == 0 : print(-1)
        else : print(queue[0])
    elif 'back' == _str[0] :
        if len(queue) == 0 : print(-1)
        else : print(queue[len(queue)-1])
    

해결 코드

import sys
from collections import deque

# import sys 로 readline() 를 해야 하고, deque 로 popleft() 를 해야 시간 초과에 걸리지 않는다.

N = int(input())
queue = deque()
for _ in range(N):
    _str = sys.stdin.readline().split()
    if 'push' == _str[0] :
        queue.append(_str[1])
    elif 'pop' == _str[0] :
        if len(queue) == 0 : print(-1)
        else : print(queue.popleft())
    elif 'size' == _str[0] :
        print(len(queue))
    elif 'empty' == _str[0] :
        if len(queue) == 0 : print(1)
        else : print(0)
    elif 'front' == _str[0] :
        if len(queue) == 0 : print(-1)
        else : print(queue[0])
    elif 'back' == _str[0] :
        if len(queue) == 0 : print(-1)
        else : print(queue[len(queue)-1])
        

📚 새롭게 알게된 내용

pop(0) 이 O(n) 의 시간 복잡도를 가지는 이유

• 0번째 요소를 pop 하고 1,2,3... n 번째 요소들을 앞으로 당겨오는 과정에서 O(n) 의 시간이 걸리기 때문입니다..!

그럼 deque 의 popeleft() 는 왜 오래 걸리지 않는가?

• collections.deque는 양쪽에서의 빠른 삽입과 삭제를 지원하는 자료구조입니다.
• 이중 연결 리스트(doubly linked list)로 구현되어 있어 맨 앞과 맨 끝에서의 삽입과 삭제가 모두 상수 시간 ( O(1) )에 이루어집니다.

[alstjr7437]

그래서 보통 저는 파이썬에서 스택 큐를 구현할 일이 있으면
list -> 스택
deque -> 큐
를 사용합니다!!
로pop(0)로 빼는 것은 스택에서 후입 선출에 제일 첫번째 요소를 뺴는 것과 같아서 O(n)이 나와서 큐를 사용할땐 덱을 사용합니다!

그리고 input을 sys를 빼먹으셨으면
전부 고치지말고 앞부분에 이 코드를 추가하시면 됩니다!

import sys
input = sys.stdin.readline

을 해놓으면 input함수가 자동으로 sys로 돌아가 시간초과가 안납니다!!!

[wkdghdwns199]

이제부터 저래 해야긋습니다..

[SeongHoonC]

deque 를 써도 계속 시간초과나서 힘들었네요..
kotlin 입출력이 오래걸려서 그랬나봐요.

BufferedWriter 로 string buffer 에 넣고 한번에 출력해 시간을 단축시켰습니당

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))
    val n = br.readLine().toInt()
    val deque = ArrayDeque<Int>()

    for (i in 0 until n) {
        val input = br.readLine().split(" ")
        if (input[0] == "push") {
            deque.add(input[1].toInt())
            continue
        }
        with(bw) {
            when (input[0]) {
                "pop" -> write(deque.removeFirstOrNull())
                "front" -> write(deque.firstOrNull())
                "back" -> write(deque.lastOrNull())
                "size" -> write(deque.size.toString())
                "empty" -> bw.write(if (deque.isEmpty()) "1" else "0")
                else -> throw IllegalArgumentException()
            }
            bw.newLine()
        }
    }
    bw.flush()
    bw.close()
}

private fun BufferedWriter.write(num: Int?) {
    if (num == null) {
        write("-1")
        return
    }
    write(num.toString())
}

[wkdghdwns199]

Kotlin 언젠간 한 번 맛 보게 될 거 같은데 무습네요