[백준] 9012번 파이썬(Python) feat.for-else문

문제

괄호 문자열(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”를 한 줄에 하나씩 차례대로 출력해야 한다. 

---

'문제 해결 아이디어'

 '()' 짝을 맞춰주기 위해 '('면 무조건 스택에 push하고 ')'면 스택이 비어있을 땐 "NO" 출력, 스택이 비지 않았을 땐 pop() 메소드를 수행한다. 이때 스택이 비지 않았다는 것은 스택 안에 '('가 요소로 들어있단 뜻이다.

따라서, 최종 스택이 비어있으면 '()'가 모두 짝을 이뤘다는 뜻이기 때문에 "YES"를 출력하고 스택이 비지 않았으면 "NO" 출력

 

'정답 코드'

t = int(input())

for _ in range(t):
    ps = input()
    stack = []

    for i in ps:
        if i == '(':
            stack.append(i)
        elif i == ')':
            if stack:
                stack.pop()
            else:
                print("NO")
                break

    else:
        if not stack:
            print("YES")
        else:
            print("NO")

 

'공부한 것'

- break : while문이나 for문을 사용하는 반복문에서 'break'를 사용하면 반복문 전체를 빠져나오게 됨

~> Q. 반복문 전체를 빠져나오는 것이 아니라 해당 조건만 건너뛰고 싶을 때는?

~> A) continue : 파이썬 프로그램이 실행되다가 'continue'를 만나면 그 아래의 코드를 수행하지 않고 반복문의 조건을 판단하는 곳으로 점프하게 됨

- for-else문 : for문이 break 등으로 중간에 빠져나오지 않고 끝까지 실행됐을 경우 else문이 실행되는 방식