백준 9012 파이썬 - 괄호 (for~ else 문)

https://www.acmicpc.net/problem/9012

9012번: 괄호

문제

괄호 문자열(Parenthesis String, PS)은 두 개의 괄호 기호인 ‘(’ 와 ‘)’ 만으로 구성되어 있는 문자열이다. 그 중에서 괄호의 모양이 바르게 구성된 문자열을 올바른 괄호 문자열(Valid PS, VPS)이라고 부른다. 한 쌍의 괄호 기호로 된 “( )” 문자열은 기본 VPS 이라고 부른다. 만일 x 가 VPS 라면 이것을 하나의 괄호에 넣은 새로운 문자열 “(x)”도 VPS 가 된다. 그리고 두 VPS x 와 y를 접합(concatenation)시킨 새로운 문자열 xy도 VPS 가 된다. 예를 들어 “(())()”와 “((()))” 는 VPS 이지만 “(()(”, “(())()))” , 그리고 “(()” 는 모두 VPS 가 아닌 문자열이다. 

여러분은 입력으로 주어진 괄호 문자열이 VPS 인지 아닌지를 판단해서 그 결과를 YES 와 NO 로 나타내어야 한다. 

입력

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 주어진다. 입력의 첫 번째 줄에는 입력 데이터의 수를 나타내는 정수 T가 주어진다. 각 테스트 데이터의 첫째 줄에는 괄호 문자열이 한 줄에 주어진다. 하나의 괄호 문자열의 길이는 2 이상 50 이하이다. 

출력

출력은 표준 출력을 사용한다. 만일 입력 괄호 문자열이 올바른 괄호 문자열(VPS)이면 “YES”, 아니면 “NO”를 한 줄에 하나씩 차례대로 출력해야 한다. 

예제 입력 1 복사

6
(())())
(((()())()
(()())((()))
((()()(()))(((())))()
()()()()(()()())()
(()((())()(

예제 출력 1 복사

NO
NO
YES
NO
YES
NO

예제 입력 2 복사

3
((
))
())(()

예제 출력 2 복사

NO
NO
NO

 

작성한코드

import sys
input= sys.stdin.readline
n = int(input())
lst = []
for _ in range(n):
  stack = []
  a = input()
  for i in a:
    if i == '(':
      stack.append(i)
    elif i ==')':
      if len(stack) !=0:
        stack.pop()
      else:
        lst.append('NO')
        break
  else:
    if len(stack) ==0:
      lst.append('YES')
    else:
      lst.append('NO')
for i in lst:
  print(i)

input으로 받은 문자가 "("면 stack 안에 append 하고, ")"일때 stack 값이 비어있으면 NO를 출력하고 비어있지 않으면 이전에 append된 "("값을 빼내온다. for 문 실행후 stack 값에 아무것도 없으면 Yes, 문자가 남아 있으면 NO를 출력한다.

 

큐 스택 관련 문제를 풀면서 for else문에 관해 공부해보았다

else 블록은 for 루프가 중간에 break로 인해 중단되지 않고 끝까지 정상적으로 실행되었을 때 실행 된다. 즉, for 루프가 완전히 실행된 후에 else 블록이 실행된다.

다른코드 예시를 보자

numbers = [1, 3, 5, 7, 9]
search_for = 4

for num in numbers:
    if num == search_for:
        print(f"{search_for}를 찾았습니다!")
        break
else:
    # 'break'로 루프가 중단되지 않았다면, 여기의 코드가 실행됩니다.
    print(f"{search_for}는 리스트에 없습니다.")

4는 리스트에 없습니다.

 

for문 아래 if문에 속해 있는 break가 발동되지 않고, else문이 실행된다.