[백준 10799번] 쇠막대기

10799번: 쇠막대기

# 괄호가 쌍을 이뤄서 pop될 때마다 +1
# 레이저인 경우 +현재 스택에 쌓인 개수

datas = list(input())

stack = []

answer = 0

for i,data in enumerate(datas):

    if data == '(':
        stack.append(data)
    elif data == ')':
        if datas[i-1] == '(': # 레이저인 경우
            answer += (len(stack)-1)
            stack.pop()
        else: # 막대기가 끝난 경우
            answer += 1
            stack.pop()

print(answer)

이번 문제는 쇠막대기와 레이저의 배치를 나타내는 괄호 표현이 주어졌을 때, 잘려진 쇠막대기 조각의 총 개수를 구하는 문제이다.

 

처음 봤을 때는 복잡해보일 수 있지만, 일단 여는 괄호가 나오면 해당 값을 스택에 담고 ()인 경우에는 현재 스택에 쌓인 값(쇠막대기 개수)를 덯주면 된다. 그리고 쇠막대기가 끝난 경우에는 막대 개수를 +1 해준 후에 스택에서 pop 해주면 된다.

 

Stack 자료구조를 활용한 간단한 문제였다 :)