문제 링크

https://programmers.co.kr/learn/courses/30/lessons/60058

문제 설명

• 균형잡힌 괄호 문자열이 주어진다
• 아래 과정을 수행

1. 입력이 빈 문자열인 경우, 빈 문자열을 반환합니다. 
2. 문자열 w를 두 "균형잡힌 괄호 문자열" u, v로 분리합니다. 단, u는 "균형잡힌 괄호 문자열"로 더 이상 분리할 수 없어야 하며, v는 빈 문자열이 될 수 있습니다. 
3. 문자열 u가 "올바른 괄호 문자열" 이라면 문자열 v에 대해 1단계부터 다시 수행합니다. 
  3-1. 수행한 결과 문자열을 u에 이어 붙인 후 반환합니다. 
4. 문자열 u가 "올바른 괄호 문자열"이 아니라면 아래 과정을 수행합니다. 
  4-1. 빈 문자열에 첫 번째 문자로 '('를 붙입니다. 
  4-2. 문자열 v에 대해 1단계부터 재귀적으로 수행한 결과 문자열을 이어 붙입니다. 
  4-3. ')'를 다시 붙입니다. 
  4-4. u의 첫 번째와 마지막 문자를 제거하고, 나머지 문자열의 괄호 방향을 뒤집어서 뒤에 붙입니다. 
  4-5. 생성된 문자열을 반환합니다.

풀이

• 그대로 구현하면 됨
• 예전에 자바로 풀었던 문제인데 파이썬으로 풀고 다시 보니 신기하게 거의 똑같이 풀었었다

코드

def solution(p):
    answer = convert(p)
    return answer


def convert(p):
    if not p:
        return p
    u, v = get_uv(p)
    if right(u):
        return u + convert(v)
    return '(' + convert(v) + ')' + reverse(u[1: -1])


def get_uv(p):
    count1, count2 = 0, 0
    for i in range(len(p)):
        if p[i] == '(':
            count1 += 1
        else:
            count2 += 1
        if count1 == count2:
            return p[:i+1], p[i+1:]
    return p, p


def right(u):
    stack = []
    for i in range(len(u)):
        if u[i] == '(':
            stack.append('(')
        else:
            if stack:
                before = stack.pop()
                if before == ')':
                    return False
            else:
                return False
    return True


def reverse(u):
    result = []
    for i in range(len(u)):
        if u[i] == '(':
            result.append(')')
        else:
            result.append('(')
    return ''.join(result)