[Seop's의 코드풀이] 백준 2504 괄호의 값 - Python

이번에 풀어볼 문제는 백준 2504번인 '괄호의 값' 이라는 문제입니다.

문제는 다음과 같습니다.

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

2504번: 괄호의 값

 

Python으로 푸는경우 +, * () 문자열을 추가시켜서 풀 수 있는 방법도 있지만 저는 스택을 사용하여 풀었습니다.

 

문제풀이 IDEA

- 소괄호, 대괄호 스택을 만들어줍니다.

- 문자열의 길이만큼의 0 배열을 만들어줍니다.(total 배열)

- "(" 일 때 소괄호 스택에 해당 인덱스를 추가해주고 '['가 나왔을때 대괄호 스텍에 해당 인덱스를 넣어줍니다.

- ')', ']'일 때 해당 괄호 스택에서 pop을 한 값( '(', '['가 나온 인덱스) 부터 현재 인덱스까지 total 배열의 합x(2 or 3)을 total[i] 에 넣어줍니다. 
  (합이 0일경우 2 or 3을 넣어줍니다)

- 결과적으로 total 배열의 합을 출력해주면 정답입니다.

 

위의 경우는 올바른 문자열 들어왔을 경우입니다. 올바르진 않은 문자열일 경우를 체크해서 0을 반환해줘야 합니다.

- 소괄호나 대괄호의 짝이 맞지 않는 경우 ex : (()

- 여는괄호보다 닫는괄호가 먼저 나오는 경우 ex. : )(][

- 소,대괄호가 닫혔는데 대,소괄가 안에여 열려있는경우 ex : ([)]

- 소,대괄호 이외의 문자가 있는경우

 

위의 경우를 잘 체크해준다면 올바르게 문제를 해결할 수 있습니다.

 

전체적인 코드는 다음과 같습니다.

질문과 피드백은 언제나 감사드립니다.

bracket = input().rstrip()

total = [0 for i in range(len(bracket))]
small = []
big = []
check = False

for i in range(len(bracket)) :
    if bracket[i] == "(" :
        small.append(i)
    elif bracket[i] == "[" :
        big.append(i)
    elif bracket[i] == ")" :
        if not small :
            check = True
            break
        st = small.pop()
        if big :
            if st < big[-1] :
                check = True
                break
        temp = 0
        for j in range(st, i) :
            temp += total[j]
            total[j] = 0
        if temp == 0 :
            total[i] = 2
        else :
            total[i] = 2*temp
    elif bracket[i] == "]" :
        if not big :
            check = True
            break
        st = big.pop()
        if small :
            if st < small[-1] :
                check = True
                break
        temp = 0
        for j in range(st, i) :
            temp += total[j]
            total[j] = 0
        if temp == 0 :
            total[i] = 3
        else :
            total[i] = 3*temp
    else :
        check = True
        break
if small or big or sum(total) == 0 or check:
    print(0)
else :
    print(sum(total))