백준 / 쇠막대기 / 10799번 / Python

*문제 출처는 백준에 있습니다.

문제 제목: 쇠막대기 / 10799번 (실버 2단계)

문제 사이트: https://www.acmicpc.net/problem/10799

 

문제 설명

---

나의 풀이

def solution():
    arr = list(input())  # 입력 문자열
    stack = []
    answer = 0

    for i in range(len(arr)):
        if arr[i] == '(':
            stack.append('(')
        else:
            stack.pop()
            if arr[i - 1] == '(':  # 레이저인 경우
                answer += len(stack)
            else:  # 쇠막대기의 끝인 경우
                answer += 1

    print(answer)


if __name__ == "__main__":
    solution()

---

※ 알아야 할 것

쇠막대기의 끝인 경우는 += 1를 해주셔야합니다!