백준 4949번] 균형잡힌 세상 코드 리뷰

문제 소스: https://www.acmicpc.net/problem/4949

4949번: 균형잡힌 세상

 

내가 작성한 코드:

list=[]

while True:
  sentence = input()
  # 종료 조건
  if (sentence=='.'):
    break
  
  list.append(sentence)

# for i in range(len(list)):
#   print(list[i])


for i in range(len(list)):
  stack=[]
  for j in range(len(list[i])):
    if list[i][j] == '(' or list[i][j]=='[':
      stack.append(list[i][j])
      
    elif list[i][j] == ')':
      # 이전에 ( 이 있다면
      if stack and stack[-1] == '(': 
        stack.pop() # ( 삭제
        
    elif list[i][j] == ']':
      # 이전에 [ 이 있다면
      if stack and stack[-1] == '[':
        stack.pop() # [ 삭제
  
  if not stack:
    print('yes')
  else:
    print('no')

 

결과 : 틀렸습니다

 

테스트 케이스는 다 맞는데, 어느 부분에서 틀린지 검토해 봐야함 -221004

 

---

원인 : 모든 문장을 입력 받은 후 처리가 아닌,

각 문장 당 입력 및 처리였음..

 

그에 따라 2차원에서 1차원 리스트로 코드를 줄였다.

 

개선 코드

 

while True:
  list = input()
  # 종료 조건
  if (list=='.'):
    break

  stack=[]
  for i in list:
    if i == '(' or i=='[':
      stack.append(i)
      
    elif i == ')':
      # 이전에 ( 이 있다면
      if stack and stack[-1] == '(': 
        stack.pop() # ( 삭제
      # 이전에 ( 이 없다면
      else:
        stack.append(i)
        
    elif i == ']':
      # 이전에 [ 이 있다면
      if stack and stack[-1] == '[':
        stack.pop() # [ 삭제
      # 이전에 [ 이 없다면
      else:
        stack.append(i)
  
  if not stack:
    print('yes')
  else:
    print('no')

 

해결 완료