📚 출처 - 4949-균형잡힌 세상
문제 설명
세계는 균형이 잘 잡혀있어야 한다. 양과 음, 빛과 어둠 그리고 왼쪽 괄호와 오른쪽 괄호처럼 말이다.
정민이의 임무는 어떤 문자열이 주어졌을 때, 괄호들의 균형이 잘 맞춰져 있는지 판단하는 프로그램을 짜는 것이다.
문자열에 포함되는 괄호는 소괄호("()") 와 대괄호("[]")로 2종류이고, 문자열이 균형을 이루는 조건은 아래와 같다.
입력
하나 또는 여러줄에 걸쳐서 문자열이 주어진다. 각 문자열은 영문 알파벳, 공백, 소괄호("( )") 대괄호("[ ]")등으로 이루어져 있으며, 길이는 100글자보다 작거나 같다.
입력의 종료조건으로 맨 마지막에 점 하나(".")가 들어온다.
출력
각 줄마다 해당 문자열이 균형을 이루고 있으면 "yes"를, 아니면 "no"를 출력한다.
입출력 예
예제 입력 | 예제 출력 | |
---|---|---|
So when I die (the [first] I will see in(heaven) is a score list). [ first in ] ( first out ). Half Moon tonight (At least it is better than no Moon at all]. A rope may form )( a trail in a maze. Help( I[m being held prisoner in a fortune cookie factory)]. ([ (([( [ ] ) ( ) (( ))] )) ]). . . | yes yes no no no yes yes |
Logic 1
처음에는 소괄호, 대괄호 두 종류로 나눠s_ch, d_ch
변수를 생성한 다음, 각각'(', '['
가 나오면 +1,')', ']'
가 나오면 -1 해주고
만약 이 값들이 음수가 되는 경우에는no
를 반환하도록 하게 만들어 봤습니다.
여기서 보충해줘야 될 부분은'('가 나온 후 ']'
가 나오는 경우,'['가 나온 후 ')'
가 나오는 경우가 부족했고, 또한 이게 맞는다면 그 전의 값을 비교해줘야 되는데 그 부분을 해주기에는 부족하다 생각했습니다.
Logic 2
Logic 1
의 로직이 허약해 리스트를 만들어 비교하는 로직으로 변경했습니다.
1. 다른 문자들은 체킹 하지 않고 소괄호와 대괄호만 체킹 합니다.
2. 여기서 시작 소괄호, 시작 대괄호가 나온다면 리스트에 추가해 줍니다.
3. 반대로 닫는 소괄호, 닫는 대괄호가 나온다면 리스트 마지막 값이 시작 소괄호, 대괄호였는지 체킹 해줍니다.
만약 나온다면 리스트의 마지막 값을pop
을 진행해 줍니다.
여기서 스택의 길이는 0일 때는 런타임 오류 발생!! 따라서 체킹 해줘야 합니다.
4. 위의 과정을 문자열이 끝날 때까지 체킹 해준 후, 리스트가 비어있다면yes
아니라면 'no`를 반환해 줍니다.
def check(words):
stack = []
for i in words:
if i == '(':
stack.append(i)
elif i == ')':
if len(stack) != 0 and stack[-1] == '(':
stack.pop()
else:
print('no')
return
elif i == '[':
stack.append(i)
elif i == ']':
if len(stack) != 0 and stack[-1] == '[':
stack.pop()
else:
print('no')
return
if len(stack) != 0:
print('no')
else:
print('yes')
while True:
words = str(input())
if words == '.':
break
check(words)