알고리즘
백준 파이썬 10799번 쇠막대기
chaewonni
2023. 11. 26. 03:31


🚀접근하기
이 문제는 처음 코드부터 작성하기 보단 혼자 메모장에 그림을 그려보면서 생각을 정리했던 것 같다.
괄호들을 보면서 스택을 써야겠다고 생각했고, "(" 이면 스택에 append하고 ")"이면 스택에서 pop하는 식으로 접근하였다.
🎉코드(시간초과)
n=input()
stack=[]
num = 0
n_list=list(n) #이거 넘넘 중요하다 제발
for i in range(len(n_list)):
if n_list[i]=="(":
stack.append(n_list[i])
elif n_list[i]==")":
if len(stack)!=0:
stack.pop()
if n_list[i-1] == "(":
for i in range(len(stack)):
num += 1
else:
num += 1
print(num)처음 작성했던 코드. string을 한 글자씩 끊어 n_list에 넣기 위해 list(n) 이라고 작성해주었다.
또한, 문제에 나왔듯이 레이저 () 와 쇠막대기 ( 열린괄호, ) 닫힌괄호 를 구분해주기 위해 n_list[i-1]=="(" 즉 앞글자가 ( 이면 레이저로, 아니면 쇠막대기로 판단하여 구분해주었다.
그런데 시간초과가 났다...! 필요없는 list와 반복문을 작성했기 때문....
🎉코드
n=input()
stack=[]
num = 0
for i in range(len(n)):
if n[i]=="(":
stack.append(n[i])
elif n[i]==")":
if len(stack)!=0:
stack.pop()
if n[i-1] == "(":
num += len(stack)
else:
num += 1
print(num)리스트를 굳이 만들지 않아도 string자체의 len을 구해주어 반복문을 돌려주면 됐다. 또한 스택에 남아있는 괄호 개수도 직접 더해주면 됐다.
💪다짐
왜 이런 시간초과가 나는 실수들을 했는지는 모르겠다만... 다음부터 최대한 불필요한 코드들을 작성하지 않도록 노력해야겠다.