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

내가 짠 코드

import sys

str = sys.stdin.readline().strip()
stk = []
arr = [[] for i in range(30)]

depth = -1
for i in range (len(str)) :
    temp = str[i]
    
    if len(stk) == 0 : #스택에 하나도 없는 경우
        stk.append(temp)
        depth += 1
    
    elif stk[-1] == '(' and temp == ')': # 스택에 () 만나는경우
        if(len(arr[depth+1]) == 0) :
            arr[depth].append(2)
        else :
            sum_val = 0
            for it in arr[depth+1] :
                sum_val += it
            arr[depth+1] = []
            arr[depth].append(2*sum_val)
        stk.pop()
        depth -= 1
    
    elif stk[-1] == '[' and temp == ']': # 스택에 [] 만나는경우
        if(len(arr[depth+1]) == 0) :
            arr[depth].append(3)
        else :
            sum_val = 0
            for it in arr[depth+1] :
                sum_val += it
            arr[depth+1] = []
            arr[depth].append(3*sum_val)
        stk.pop()
        depth -= 1
    
    else :
        stk.append(temp)
        depth += 1

    # print(depth)
    # print(arr)
    
if len(stk) != 0 :
    print(0)
else :
    sum_val = 0
    for it in arr[0] :
        sum_val += it
    print(sum_val)

• 이차원 배열을 만들어서 풀었다.
• 괄호가 시작하면 depth가 증가한다.
• 괄호가 종료되면,
  • 다음 depth에 요소가 있는 경우 그 값을 전부 더한 후, ()면 2, []면 3을 곱해 현재 depth에 넣는다.
  • 없는 경우 현재 depth에 3 넣는다.
  • 이렇게 풀긴 하였으나 복잡하기도 하고, 현재 문제처럼 최대 depth가 30 정도면 별 문제가 없겠으나, 더 많아진다면 문제가 생길 것으로 보인다.

다른 사람이 짠 코드

bracket = list(input())

stack = []
answer = 0
tmp = 1

for i in range(len(bracket)):

    if bracket[i] == "(":
        stack.append(bracket[i])
        tmp *= 2

    elif bracket[i] == "[":
        stack.append(bracket[i])
        tmp *= 3

    elif bracket[i] == ")":
        if not stack or stack[-1] == "[":
            answer = 0
            break
        if bracket[i-1] == "(":
            answer += tmp
        stack.pop()
        tmp //= 2

    else:
        if not stack or stack[-1] == "(":
            answer = 0
            break
        if bracket[i-1] == "[":
            answer += tmp

        stack.pop()
        tmp //= 3

if stack:
    print(0)
else:
    print(answer)

• tmp, ans 두 개의 변수를 이용한다.
• 괄호가 ( 혹은 [ 로 시작될 경우
  • 이 안에 있는 수들은 전부 2 혹은 3으로 곱해진단 의미다.
  • tmp에 해당 수를 곱해준다. 괄호가 중첩될수록 계속 이 값을 곱해줄 것이다.
  • 스택에 해당하는 것을 push한다.
• 괄호가 ) 혹은 ] 로 끝날 경우
  • 만약 스택의 마지막 값이 ( 인데 ] 로 끝나거나, 그 반대의 경우 잘못된 괄호이다.
    • → 0이 출력된다.
  • 스택의 마지막 값이랑 맞는 괄호로 끝났을 경우
    • 입력 받은 리스트를 확인해본다 (스택이 아니다). 만약 현재 요소 앞이 현재 요소와 맞는 괄호라면,
      • 여태껏 곱해진 tmp를 ans에 더해준다.
    • 리스트 상, 현재 요소 앞이 현재 요소와 맞는 괄호가 아니라면,
      • 그냥 넘어간다.
  • 괄호가 끝났으므로 스택에서 pop해주고, 곱했던 tmp를 예전 값이 되도록 나눠준다.
• 이를 반복한다.

참고

https://hongcoding.tistory.com/114