[백준 2504번] 괄호의 값 (python/파이썬)

조건을 엄청 추가해서 푼 구현 문제. 코드가 너무 더러운데 더 좋은 풀이 방법이 있을 듯

 

🌼 문제 링크

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

2504번: 괄호의 값

• 티어: 실버 I
• 분류: 구현, 자료구조, 스택, 재귀

 

 

❗ TRIAL 1.

코드

inputs = list(input())

couple = {'(': ')', '[': ']'}
value = {'(': 2, '[': 3}

while len(inputs) != 1:
    ctr = 0
    new_inputs = []
    while ctr < len(inputs)-1:
        v = inputs[ctr]
        w = inputs[ctr+1]
        if v == '(' or v == '[':
            if w == couple[v]: #() or []: change into number
                ctr += 2
                new_inputs.append(value[v])
            elif type(w) == int: #(w) or {w}: change into 2/3*w
                x = inputs[ctr+2]
                if x == couple[v]:
                    ctr += 3
                    new_inputs.append(value[v]*w)
                else:
                    ctr += 1
                    new_inputs.append(v)
            else:
                ctr += 1
                new_inputs.append(v)
        elif type(v) == int:
            if type(w) == int:
                ctr += 2
                new_inputs.append(v+w)
            else:
                ctr += 1
                new_inputs.append(v)
        else:
            ctr += 1
            new_inputs.append(v)
    if ctr == len(inputs)-1:
        new_inputs.append(inputs[ctr])
    if len(inputs) == len(new_inputs):
        inputs = [0]
    else:
        inputs = new_inputs

print(inputs[0])

 

틀린 원인

 

런타임 에러 (IndexError). 매번 현재 node와 다음 node만 비교하다가 (X) 형태일 경우를 찾기 위해 다다음 node까지 검사하는 경우가 있었는데, 이 때 (X에서 값이 끝나는 경우에 index error가 발생.

예외 케이스: https://www.acmicpc.net/board/view/68189

 

 

💡 Solution

코드

inputs = list(input())

couple = {'(': ')', '[': ']'}
value = {'(': 2, '[': 3}

while len(inputs) > 1 or type(inputs[0]) != int:
    ctr = 0
    new_inputs = []
    while ctr < len(inputs)-1:
        v = inputs[ctr]
        w = inputs[ctr+1]
        if v == '(' or v == '[':
            if w == couple[v]: #() or []: change into number
                ctr += 2
                new_inputs.append(value[v])
            elif type(w) == int: #(w) or {w}: change into 2/3*w
                try:
                    x = inputs[ctr+2]
                except:
                    inputs = [0]
                    break
                if x == couple[v]:
                    ctr += 3
                    new_inputs.append(value[v]*w)
                else:
                    ctr += 1
                    new_inputs.append(v)
            else:
                ctr += 1
                new_inputs.append(v)
        elif type(v) == int:
            if type(w) == int:
                ctr += 2
                new_inputs.append(v+w)
            else:
                ctr += 1
                new_inputs.append(v)
        else:
            ctr += 1
            new_inputs.append(v)
    if ctr == len(inputs)-1:
        new_inputs.append(inputs[ctr])
    if len(inputs) == len(new_inputs):
        inputs = [0]
    else:
        inputs = new_inputs

print(inputs[0])

 

코드 설명

• 재귀적으로 간단한 경우부터 숫자로 바꿔 줌.
• () : 숫자로 변경
• (X): X*2로 변경. (이 경우에 만약 IndexError가 발생하면, 마지막 괄호가 나오기 전에 문자열이 끝난 것이므로 문자열에 오류가 있는 것. 바로 0 반환.)
• XY: X+Y로 변경
• [에 대해서도 동일하게 적용
• 만약 끝까지 다 돌았는데도 길이가 바뀌지 않았다면 규칙에 부합하는 경우가 없는 경우. 이 경우 문자열에 오류가 있는 것이므로 바로 0 반환.
• 문자열이 정수 하나만 남을 때까지 반복.