[python] 쇠막대기

문제 출처

쇠막대기

 

풀이

def solution(arrangement):
    answer = 0
    pipe = 1
    stack = ''
    for i in range(1, len(arrangement)):
        if arrangement[i] == '(':
            pipe += 1
        else:
            pipe -= 1
            if arrangement[i-1] == '(':
                answer += pipe
            else:
                answer += 1
    return answer