[Python]프로그래머스 - 쇠막대기(level2)

문제출처 - https://programmers.co.kr/learn/courses/30/lessons/42585

코딩테스트 연습 - 쇠막대기 | 프로그래머스

def solution(arrangement):
    answer = 0
    stack = []
    
    # ()일때만 레이저
    
    for i in range(len(arrangement)):
        if arrangement[i] == '(':
            stack.append('(')
        else:
            stack.pop()
            # 레이저인 경우
            if arrangement[i-1] == '(':
                answer += len(stack)
            else:  # 막대기가 끊긴 경우
                answer += 1            
    
    return answer

설명

 

왼쪽괄호 '('가 나온 경우 스택에 추가

오른쪽괄호 ')'가 나온 경우 일단 스택에서 한개 빼주기

   -> 레이저인 경우(바로 전의 괄호가 '('였다면 레이저라는거 ㅇㅇ), 이때는 answer에 현재 스택의 길이만큼 더해주기

   -> 막대기가 끊긴 경우, 예를들어 ))가 연속으로 나온 경우, answer에 1만 추가