글 읽기 - 파이썬) 2504번 질문입니다

kso626 3년 전 0

예제나 기본적인 몇가지 더해서 넣어봐도 정상적으로 출력이 되는데

제출을하게되면 틀렸다고 나와서 질문드립니다.

class Stack():
    def __init__(self):
        self.str_list = []

    def push(self, str):
        self.str_list.append(str)

    def pop(self):
        if len(self.str_list) != 0:
            return self.str_list.pop()
        else:
            return -1

    def size(self):
        return len(self.str_list)

    def empty(self):
        if len(self.str_list) == 0:
            return 1
        else:
            return 0

    def top(self):
        if len(self.str_list) != 0:
            return self.str_list[-1]
        else:
            return -1

stack = Stack()
count = 0
for i in list(input()):
    if i == '(' or i == '[':                                        # 입력이 '(' 또는 '[' 라면 스택에 push
        stack.push(i)
    elif stack.top() == '(':
        if i == ')':                                                # stack.top() 이 '(' 이고 입력이 ')'일때
            stack.pop()                                             # 열린괄호 제거
            count = 2
            if type(stack.top()) == int and stack.top() != -1:      # stack.top()이 숫자일때
                count += stack.pop()                                # 두 수를 더한다
                stack.push(count)
            else:
                stack.push(count)
        else:
            count = 0                                               # stack.top() 이 '(' 일때 입력이 ']'라면 잘못된 문이다.
            break
    elif stack.top() == '[':                                        # 위와 동일하지만 괄호문만 '()' -> '[]' 로 바뀜
        if i == ']':
            stack.pop()
            count = 3
            if type(stack.top()) == int and stack.top() != -1:
                count += stack.pop()
                stack.push(count)
            else:
                stack.push(count)
        else:
            count = 0
            break
    else:                                                           # stack.top()가 숫자인 경우
        count = stack.pop()
        if i == ')':
            if stack.top() == '(':                                  # 위와 동일하지만 괄호 내부에 괄호문은 곱셈처리
                stack.pop()
                count *= 2
                if type(stack.top()) == int and stack.top() != -1:
                    count += stack.pop()
                    stack.push(count)
                else:
                    stack.push(count)
            else:
                count = 0
                break
        elif i == ']':
            if stack.top() == '[':
                stack.pop()
                count *= 3
                if type(stack.top()) == int and stack.top() != -1:
                    count += stack.pop()
                    stack.push(count)
                else:
                    stack.push(count)
            else:
                count = 0
                break
print(count)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

파이썬) 2504번 질문입니다

kso626 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회 / 고등학교 대회

출처

대학교 대회

도움말