[BOJ] 10799 쇠막대기 (Python / 파이썬)

🧷 문제

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

입력으로 주어진 괄호의 쌍을 분석하여 쇠막대기가 레이저에 의해 몇 개의 조각으로 잘려지는지 알아보는 문제이다.

🛠 풀이

이 문제는 스택을 이용하여 해결할 수 있는 문제이다.

Step 1.
입력으로 주어진 쇠막대기와 레이저의 배치를 pipes로 받고 for문을 이용해 앞에서부터 탐색을 시작한다.
(를 만나면 stack에 push해준다.

Step 2.
)를 만나면 pipes의 바로 전 인덱스를 살펴보고 만약 pipes[i-1]이 (라면 레이저이기 때문에 stack에서 pop해주고 res에 stack의 길이만큼 더해준다.
그렇지 않고 pipes[i-1]이 )라면 해당 라인의 막대기가 끝나는 지점이기 때문에 res에 1을 더해준다.

처음에 문제를 마주했을 때는 어떻게 해결해야 할지 전혀 감을 못잡았던 것 같다.
코딩테스트에서는 이보다 훨씬 복잡한 문제가 나올게 당연하기 때문에 문제이해능력을 높여야 할 필요성을 느낄 수 있었다.

🖊 나의 코드

import sys

input = sys.stdin.readline

if __name__ == "__main__":
    pipes = input().rstrip()
    stack = []
    res = 0

    for i in range(len(pipes)):
        if pipes[i] == '(':
            stack.append(pipes[i])

        else:
            stack.pop()
            if pipes[i-1] == '(':
                res += len(stack)
            else:
                res += 1

    print(res)