백준 9012번. 괄호 (Python / 파이썬)

문제

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

---

문제 풀이

괄호 문제는 '(' 와 ')'만으로 구성된 문자열이 있을 때

올바른 괄호 문자열인지 판단하는 문제입니다.

 

올바른 괄호 문자열이 되려면 두 가지 조건을 만족해야 합니다.

- '(' 의 개수와 ')' 의 개수가 같아야 함

- '(' 와 ')' 가 서로 짝을 이뤄야 함

 

따라서, count() 함수를 이용해 '('와 ')'의 개수를 각각 세어주고

만약 개수가 같지 않다면 result를 False로 만들어주었습니다.

만약 개수가 같다면 올바른 괄호 문자열인지 검사하도록 해주었습니다.

 

올바른 괄호 문자열인지 검사하는 함수 check_proper()를 만들어주었습니다.

만약 '(' 이라면 count 값을 증가시키고 그렇지 않다면 count 값을 감소시킵니다.

이때, 값을 감소시키기 전 만약 count 값이 0이라면 False를 return해줍니다.

올바른 괄호 문자열이라면 count를 세는 과정에서 0보다 작아지지 않습니다.

따라서 이 과정을 모두 수행했다면 True를 return해줍니다.

 

위의 과정을 수행하고 나면 result에는 

올바른 괄호 문자열이라면 True

그렇지 않다면 False가 저장되어 있을 것입니다.

따라서 result가 True라면 YES

False라면 NO를 출력해주었습니다.

---

My Code

# 올바른 괄호 문자열인지 검사
def check_proper(s):
    count = 0
    for i in s:
        if i == '(': # '('를 만나면 count 1증가
            count += 1
        else:        # ')'를 만나면 count 1감소
            if count == 0: # 만약 0이라면
                return False # 올바른 괄호 문자열이 아니므로 False를 반환
            count -= 1
    return True # 올바른 괄호 문자열이므로 True 반환

for tc in range(int(input())): 
    s = input() # 괄호 문자열
    a = s.count('(') # '('의 개수
    b = s.count(')') # ')'의 개수
    if a == b: # 개수가 같다면
        result = check_proper(s) # 올바른 괄호 문자열인지 판단
    else:      # 개수가 다르다면
        result = False # 올바른 괄호 문자열이 아니므로 False 저장

    if result: 
        print('YES')
    else:
        print('NO')