Data Structures & Algorithms

[Algorithms] 스택,큐,해쉬,힙-쇠막대기(스택)

숄구-ml 2022. 5. 17. 21:47

 

 

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import sys
 
sys.stdin = open("input""rt")
code=input()
 
stack=[]
piece=0
for i in range(len(code)):
    if code[i]=='(':                # '(' 면 stack에 쌓아주고
        stack.append(code[i])
    else:
        if code[i-1]=='(':          # '()' 면 레이저 발사
            stack.pop()             # 스택에 쌓인 막대 개수만큼 조각이 난다
            piece+=len(stack)
        else:
            stack.pop()             # '))' 면 막대 하나가 끝남
            piece+=1                # 따라서 마지막의 1조각 더해준다
 
print(piece)
cs
728x90