백준 알고리즘 풀이 : 10799번 쇠막대기 python

 

 

📌 문제

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

10799번: 쇠막대기

 

---

 

🔖 풀이

 

처음 문제를 보고 바로 이해하지 못했다.

풀이는 어렵지 않지만 스택을 이용해서

풀어야겠다고 생각하는 문제 이해는 어려운 것 같다.

 

 

bar = list(input())

answer = 0
stack = []

for i in range(len(bar)):
  # i가 "("인 경우
  if bar[i] == "(":
    stack.append(bar[i])
  # i가 ")"인 경우
  else:
    # "()"라면 레이저 파트
    if bar[i-1] == "(":
      stack.pop()
      answer+= len(stack)
    # "))"라면 쇠막대기 끄트머리 표현
    else:
      stack.pop()
      answer += 1

print(answer)

 

총 3가지 조건문을 만들어 주면 된다.

 

1. i가 "("인 경우

2. i가 ")"인 경우

  2-1. "()"인 경우

  2-2. "))"인 경우

 

 

"()"는 레이저이기 때문에 스택에서 "("를 하나 빼주고

스택에 남아있는 "("를 answer에 더해준다.

 

"))"라면 쇠막대기의 끄트머리 표현이기 때문에

하나가 나올때마다 하나씩 추가해주면 된다.