[백준 9012: 괄호] 파이썬 풀이 _ 특정 문자열 제거, 변경 replace()

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

 

---

💻 코드 및 해설

• “()”를 한 세트로 해서 제거 ⇒ replace 함수 사용 : ‘()’ 문자열을 빈 문자열 ''로 대체
• 반복문
  문자열이 빈 문자열이 되거나 (괄호가 모두 제거되어 YES인 상태), 대체했는데도 원래 상태와 똑같으면 (더이상 제거할 ()가 없는데 빈 문자열이 안된거니까 NO인 상태) → 반복문 탈출
• 계속 돌면서 문자열 ps를 제거한 결과로 대체 → 마지막 결과가 빈 문자열이 되었으면 YES 출력 → 괄호들이 남아있으면 NO 출력

import sys
input = sys.stdin.readline

t = int(input()) #테스트 케이스의 수 t

for _ in range(t):
  ps = input().strip()
  
  while (ps != '' and ps.replace('()', '') != ps):
    ps = ps.replace('()', '')
  if (ps == ''):
    print("YES")
  else:
    print("NO")

---

replace() 

문자열 안에서 특정 문자를 새로운 문자로 변경하는 함수

 

변수.replace(old, new, [count])

• old: 현재 문자열에서 변경하고 싶은 문자
• new: 새로 바꿀 문자
• count: 변경할 횟수
  횟수를 입력하지 않으면 old의 문자열 전체를 변경, 기본값 count = -1

📍 참고 링크

파이썬 replace( ) 문자열을 변경하는 함수 (Python)