[Python/스택,큐] 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”를 한 줄에 하나씩 차례대로 출력해야 한다. 

입력 예제	출력 예제
4 (())()) (( ())(() (()())	NO NO NO YES

---

풀이

아이디어:

1. (의 개수를 구하는 변수와 )의 개수를 구하는 변수를 따로 쓰지 말고

+1 , -1로 (와 )의 개수가 같은 지 확인

2. deque 안 쓰고 list로 사용 가능

num = int(input())
for n in range(num):
    arr = list(input()) #입력받은 string 값 -> list로 변환해줘야함
    # 1차 필터링(())
    if arr.count('(') != arr.count(')'):
        print("NO")
        continue
    
    temp = 0
    for a in arr:
        if a == "(":
            temp += 1
        elif a == ")":
            temp -= 1
        if temp < 0:
            print("NO")
            break
                
    if temp > 0:
        print("NO")
    if temp == 0:
        print("YES")

참고

 

 

다른 풀이

스택 활용.  스택을 '('만 저장하는 새로운 list라고 생각

문자열을 한 글자씩 검사해서 '('이면 stack에 넣음. 

')'이면 stack에 값이 있는 지 확인. 값이 없다면 )의 개수가 (의 개수보다 많은것이므로 NO

             stack에 값이 있다면 pop()해줌. (문자열 다 검사하고 나면 stack에 아무 것도 없어야 YES)

num = int(input())
for _ in range(num):
    arr = list(input())
    stack = [] # 가장 최근에 들어간 (를 빼기 위해
    flag = 0
    
    for a in arr:
        if a == "(":
            stack.append(a)
        elif a== ")": 
            if stack: #stack이 비어있지 않다면
                stack.pop()
            elif not stack: #stack이 비어있다면
                print("NO")
                flag = 1
                break
    if stack and flag == 0: #stack이 비어있지 않다면
        print("NO")
    elif not stack and flag == 0:
        print("YES")

참고