(3주차) 2800: 괄호 제거

✏ 풀이

from itertools import combinations

exp = list(input())
brk_idx = list()  # 튜플(여는괄호idx, 닫는괄호idx)을 담는 리스트
left_brk_idx = list() # 여는 괄호를 잠시 저장해놓는 스택
result = list()  # 결과 문자열을 리스트에 저장

for i in range(len(exp)):
  if exp[i] == ')': # 닫는 괄호면
    brk_idx.append((left_brk_idx.pop(), i))
  elif exp[i] == '(':  # 여는 괄호면
    left_brk_idx.append(i)
    
for i in range(1, len(brk_idx)+1):
  for x in combinations(brk_idx, i):
    # print("x", x)
	    temp = exp[:]  # 주의: temp = exp 라고 하면 reference가 넘어감
		# 여기서 지워서 result에 저장    
    for l, r in x:
      temp[l] = ''
      temp[r] = ''
    result.append(''.join(temp))
result = list(set(result))  # 괄호가 여러개 중첩돼있다면 ex) (((1)))(2) 조합을 돌려도 같은 결과가 나올 수 있음! 중복제거해주자
for x in sorted(result):
  print(x)

• 여는 괄호와 닫는 괄호의 인덱스 쌍을 튜플 형태로 리스트에 저장해서 풀었다.
• 조합을 이용해 지울 괄호쌍을 정하고, 조합의 결과에 따라 만들어진 문자열을 result 리스트에 저장해 주었다.
  • 조합은 아래 1주차 정리 참고!!
    • itertools 라이브러리의 permutations(순열), combination(조합) 클래스
    • for문 + combination(조합) 최적화: 2차원 리스트/튜플에서 안쪽 원소에 접근
• 사전식으로 출력하라고 했기 때문에 sorted 메서드도 사용하였다!
• exp와 exp[:]의 차이
  • exp와 exp[:]의 차이
• 처리한 반례
  • 처리한 반례

(3주차) 9012: 괄호

✏ 풀이

import sys

input = sys.stdin.readline

def isVPS(str):
  stack = list()  # stack에는 여는 괄호 '('만 담는다.
  for x in str:
    if x == "(":
      stack.append(x)
    else:  # 닫는 괄호 ')'를 만나면 stack에 있는 여는 괄호를 pop한다.
      try:
        stack.pop()
      except:
        return False
  if not stack:  # 스택이 비어있으면
    return True
  else:
    return False
        
t = int(input())
for _ in range(t):
  a = list(input().rstrip())
  if isVPS(a):
    print("YES")
  else:
    print("NO")

문자열이 VPS인지 확인하고 boolean값을 리턴하는 isVPS()함수를 작성했다.