글 읽기 - [파이썬] input시에 rstrip()을 꼭 사용해야합니까?

hyeon0528 2년 전 0

틀렸습니다 떠서 게시판 반례들 살피다가

input받을 때 , rstrip()하지 않으면 틀렸습니다가 뜨고

rstrip()을 써주면 맞습니다.

테케에 공백이 포함되어있는것 같습니다.

'''
https://www.acmicpc.net/problem/2504
() = 2
[] = 3
(x) = 2*x
[x] = 3*x
(xy) = (x)+(y)
'''

import sys
input = sys.stdin.readline

#입력시 rstrip()하지 않으면 95%에서 틀렸습니다. 뜸.
brakets = input().rstrip()
stack = []
for i in range(len(brakets)):
    flag = True
    if brakets[i] == '(' or brakets[i] == '[':
        stack.append(brakets[i])
    elif brakets[i] == ')' and stack:
        if stack[-1] == '(':
            stack.pop()
            stack.append('2')
        elif stack[-1] == '[':
            flag=False
        elif stack[-1].isdigit():
            sum = 0
            while stack:
                if stack[-1].isdigit():
                    sum += int(stack.pop())
                elif stack[-1] == '(':
                    stack.pop()
                    sum *=2
                    stack.append(str(sum))
                    break
                elif stack[-1] == '[':
                    flag = False
                    break
    elif brakets[i] == ']' and stack:
        if stack[-1] == '[':
            stack.pop()
            stack.append('3')
        elif stack[-1] == '(':
            flag = False
        elif stack[-1].isdigit():
            sum = 0
            while stack:
                if stack[-1].isdigit():
                    sum+= int(stack.pop())
                elif stack[-1] == '[':
                    stack.pop()
                    sum *=3
                    stack.append(str(sum))
                    break
                elif stack[-1] == '(':
                    flag = False
                    break
    else:
        flag = False
    if flag == False:
        print(0)
        break
    print(stack)
if flag == True:
    try:
        nums = list(map(int,stack))
    except:
        print(0)
    else:
        result = 0
        for num in nums:
            result += num
        print(result)