PS/Python
[Python] 백준 #4949 균형잡힌 세상 - stack
yoo.o
2020. 8. 13. 15:18
반응형
문제
4949번: 균형잡힌 세상
문제 세계는 균형이 잘 잡혀있어야 한다. 양과 음, 빛과 어둠 그리고 왼쪽 괄호와 오른쪽 괄호처럼 말이다. 정민이의 임무는 어떤 문자열이 주어졌을 때, 괄호들의 균형이 잘 맞춰져 있는지 판단
www.acmicpc.net
풀이
괄호 문제라 스택을 이용해서 풀었다.
stack = []
while True:
stack.clear()
flag = 0
line = input()
if line == ".":
break
for each in line:
if flag == 1:
break
if each == "(" or each == "[":
stack.append(each)
elif each == ")":
if len(stack) > 0 and stack[-1] == "(":
stack.pop()
else:
flag = 1
elif each == "]":
if len(stack) > 0 and stack[-1] == "[":
stack.pop()
else:
flag = 1
if len(stack) == 0 and flag == 0:
print("yes")
else:
print("no")yes/no로 출력해야하는데 YES/NO로 짜서 자꾸 틀렸다고 나와서 삽질함 ..
반응형