백준 파이썬 10866 덱

🚀접근하기

이 문제는 그냥 덱을 활용할 수 있냐를 묻는 문제라 별로 어렵지 않았다.

push_back X 에서 X의 값은 항상 달라지기 때문에 이를 어떻게 코드로 구현할 지 먼저 생각해보았던 것 같다.

 

🎉코드(시간초과)

from collections import deque

d = deque()
n=int(input())

for i in range(n):
    s=input()
    s_list=s.split(" ")
    if s_list[0] == 'push_back':
        d.append(s_list[1]) #덱의 뒤에 추가
    elif s_list[0] == 'push_front':
        d.appendleft(s_list[1]) #덱의 앞에 추가
    elif s_list[0] == 'pop_front':
        if len(d) == 0:
            print("-1")
        else:
            print(d.popleft())
    elif s_list[0] == 'pop_back':
        if len(d) == 0:
            print("-1")
        else:
            print(d.pop())
    elif s_list[0] == 'size':
        print(len(d))
    elif s_list[0] == 'empty':
        if len(d) == 0:
            print("1")
        else:
            print("0")
    elif s_list[0] == 'front':
        if len(d) == 0:
            print("-1")
        else:
            print(d[0])
    elif s_list[0] == 'back':
        if len(d) == 0:
            print("-1")
        else:
            print(d[-1])

 

push_back X 로 정수 X를 받으면 공백 단위로 split 시켜 리스트에 push_back, X(정수)를 따로 넣어주고, list[0]의 값이 뭐냐에 따라 if, else 문을 작성해주었다. 이렇게 하면 특이한 케이스인 push_back X와 push_front X에서 X의 값이 list[1]에 저장되어 있기 때문에 list[1] 즉, 원하는 정수를 덱의 앞뒤에 넣어줄 수 있다.

 

그런데.. 시간초과가 났다. 아무래도 'input()'함수를 사용하여 입력을 받을 때 매번 호출할 때마다 입력 대기 상태에 있어서 생기는 문제같다.

 

🎉코드

from collections import deque
import sys

d = deque()
n=int(sys.stdin.readline().strip())

for i in range(n):
    s= sys.stdin.readline().strip()
    s_list=s.split(" ")
    if s_list[0] == 'push_back':
        d.append(s_list[1]) #덱의 뒤에 추가
    elif s_list[0] == 'push_front':
        d.appendleft(s_list[1]) #덱의 앞에 추가
    elif s_list[0] == 'pop_front':
        if len(d) == 0:
            print("-1")
        else:
            print(d.popleft())
    elif s_list[0] == 'pop_back':
        if len(d) == 0:
            print("-1")
        else:
            print(d.pop())
    elif s_list[0] == 'size':
        print(len(d))
    elif s_list[0] == 'empty':
        if len(d) == 0:
            print("1")
        else:
            print("0")
    elif s_list[0] == 'front':
        if len(d) == 0:
            print("-1")
        else:
            print(d[0])
    elif s_list[0] == 'back':
        if len(d) == 0:
            print("-1")
        else:
            print(d[-1])

 

 이런 상황에서는 입력을 미리 받아두고 처리하는 방식을 사용하여 시간을 줄일 수 있다!

import sys
n=int(sys.stdin.readline().strip())

코드를 사용하여 입력을 미리 다 받아둔 후 처리하도록 하였다

 

 

💪다짐

시간을 단축하기 위해 input 말고도 sys.stdin.readline()을 사용하여 빠르게 입력을 받을 수 있는 방법을 사용해야겠다!