글 읽기 - python 틀린이유를 모르겠습니다... 고수님들 도와주세요..

kpeel5839 2년 전 0

큐를 구현해서 풀고 싶어서 클래스로 만들어서 풀었습니다.
일단은 예제는 답이 정확하게 나오는데 제출하면 바로 틀렸다고 나옵니다... 어디가 문제일까요?
import sys
class Node: #노드
    def __init__(self,value):
        self.value = value
        self.next_node = None
        self.prev_node = None
class Queue: #큐
    def __init__(self): #생성자
        self.root_node = None #큐의 루트노드
        self.last_ndoe = None #큐의 맨 마지막 노드
        self.size = 0 #사이즈를 O(1)시간에 구하기 위해 선언한 변수
    def push_front(self, value):
        if self.root_node == None: #큐가 비어있을 경우
            self.root_node = Node(value) #큐에다가 노드 삽입
            self.last_node = self.root_node #라스트 노드도 갱신
            self.size += 1 #하나 추가 했으니 사이즈도 증가
            return
        #큐가 비어있지 않을 경우
        self.root_node.prev_node = Node(value) #푸쉬 프론트이기 때문에 맨앞에다가 넣기 위해서 현 루트노드의 이전 노드에 삽입
        self.root_node.prev_node.next_node = self.root_node #그리고 삽입한 노드의 next_node를 현 루트노드로 지정
        self.root_node = self.root_node.prev_node #루트노드 갱신
        self.size += 1 #사이즈 증가
    def push_back(self,value):
        if self.root_node == None: #큐가 비어있을 경우 (위와 동일함)
            self.root_node = Node(value) 
            self.last_node = self.root_node
            self.size += 1
            return
        #큐가 비어있지 않은 경우
        self.last_node.next_node = Node(value) #라스트 노드의 next_node에 노드 추가
        self.last_node.next_node.prev_node = self.last_node #삽입한 노드의 이전 노드를 현 라스트 노드로 지정
        self.last_node = self.last_node.next_node #라스트 노드 갱신
    def pop_front(self): #루트노드를 빼기
        if self.root_node == None: #큐가 비어있을 경우 -1을 출력
            sys.stdout.write(str(-1) + '\n')
            return
        if self.root_node.next_node == None: #큐의 사이즈가 1인 경우
            sys.stdout.write(str(self.root_node.value) + '\n') #일단 먼저 출력
            self.root_node = None #루트노드를 None으로 갱신
            self.last_node = self.root_node #라스트 노드도 똑같이
            self.size -= 1 #사이즈 감소
            return
        #큐가 비어있지도 사이즈가 1도 아닌 경우
        sys.stdout.write(str(self.root_node.value) + '\n') #일단 출력
        self.root_node = self.root_node.next_node #루트노드를 갱신
        self.root_node.prev_node = None #갱신한 루튼노드의 이전 노드를 None으로 하면서 이전 루트노드 삭제
        self.size -= 1 #사이즈 감소
    def pop_back(self): #라스트 노드 빼기
        if self.root_node == None: #큐가 비어있을 경우 -1을 출력
            sys.stdout.write(str(-1) + '\n')
            return
        if self.root_node.next_node == None: #큐의 사이즈가 1일 경우 (위와 동일하게 작동)
            sys.stdout.write(str(self.root_node.value) + '\n')
            self.root_node = None 
            self.last_node = self.root_node
            self.size -= 1
            return
        #큐가 비어있지도 사이즈가 1도 아닌 경우
        sys.stdout.write(str(self.last_node.value) + '\n') #일단 먼저 출력
        self.last_node = self.last_node.prev_node #라스트노드를 갱신
        self.last_node.next_node = None #갱신된 라스트 노드의 next_node를 None으로 갱신함으로서 이전 라스트노드 삭제
        self.size -= 1 #사이즈 감소
    def print_size(self): #사이즈 출력
        sys.stdout.write(str(self.size) + '\n')
    def empty(self): #비어있는지 출력
        if self.root_node == None: #비어있을 경우
            sys.stdout.write(str(1) + '\n')
        else: #비어있지 않은 경우
            sys.stdout.write(str(0) + '\n')
    def front(self): #앞에 것을 출력
        if self.root_node == None: #비어있을 경우 -1을 출력
            sys.stdout.write(str(-1) + '\n')
        else: #비어있지 않을 경우에는 루트 노드의 value 출력
            sys.stdout.write(str(self.root_node.value) + '\n')
    def back(self): #뒤에 것을 출력
        if self.root_node == None: #비어있는 경우에는 -1을 출력
            sys.stdout.write(str(-1) + '\n')
        else: #비어있지 않은 경우에는 라스트 노드의 value를 출력
            sys.stdout.write(str(self.last_node.value) + '\n')
size = int(sys.stdin.readline()) #명령 횟수 입력
st_list = [[0 for _ in range(2)] for _ in range(size)] #명령을 받을 문자열 리스트
index = 0 #st_list 의 인덱스를 선언
queue = Queue() #큐를 생성
for i in range(size): #명령어 한 줄씩 입력 받기
    st = sys.stdin.readline() #명령 한 줄 입력 받기

    for j in range(len(st)): #명령 해석
        if st[j] == '\n': #sys.stdin.readline()을 써서 줄바꿈 문자는 받지 않기
            continue
        if st[j] == ' ': #공백을 만나면 st_list[1]에다가 입력 받기
            index += 1 #인덱스 증가
            st_list[1] = '' #st_list[1]을 새로 받을 준비
            continue
        if j == 0: #새로운 명령이 들어온 것이니 새로 처리
            index = 0 #인덱스 0으로 다시 초기화
            st_list[0] = '' #위와 마찬가지로 st_list[0]도 새로 받을 준비
            st_list[0] += st[j] #그 와 동시에 첫 번째 문자 받기
            continue
        st_list[index] += st[j] #위의 상황이 아닌 경우에는 그냥 st_list[index] 에 입력 받기
        
    #입력 받고 명령에 알맞게 큐의 함수들을 실행하는 구문들
    if st_list[0] == 'push_front': 
        queue.push_front(int(st_list[1]))
    elif st_list[0] == 'push_back':
        queue.push_back(int(st_list[1]))
    elif st_list[0] == 'pop_front':
        queue.pop_front()
    elif st_list[0] == 'pop_back':
        queue.pop_back()
    elif st_list[0] == 'size':
        queue.print_size()
    elif st_list[0] == 'empty':
        queue.empty()
    elif st_list[0] == 'front':
        queue.front()
    elif st_list[0] == 'back':
        queue.back()
댓글을 작성하려면 로그인해야 합니다.
문제

문제

출처

ICPC

python 틀린이유를 모르겠습니다... 고수님들 도와주세요..

kpeel5839 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말
