[Python] 백준 10799. 쇠막대기

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

10799번: 쇠막대기

2019년 어느 기업 코딩테스트에선가 이 문제를 봤었던 걸로 기억한다.

 

stack을 사용하되, 이전 state를 저장하는 prev 변수를 두어 '레이저인지 / 쇠막대기인지' 구분했다.

prev == '('이고 현재 값이 ")"이면 레이저, prev == ')'이고 현재 값이 ")"이면 쇠막대기로 볼 수 있다.

 

 

	import sys
	arr = list(sys.stdin.readline())
	stack = []
	answer = 0
	prev = None
	for i in range(len(arr)):
	if arr[i] == '(':
	stack.append(arr[i])
	elif prev == '(' and arr[i] == ')':
	stack.pop()
	answer += len(stack)
	elif prev == ')' and arr[i] == ')':
	answer += 1
	stack.pop()
	prev = arr[i]
	print(answer)

view raw [백준] 쇠막대기.py hosted with ❤ by GitHub