프로그래머스 [lv2] 괄호변환 파이썬

접근

복잡해 보이지만, 문제의 요구조건을 차례로 구현 하면 해결할 수 있습니다.. 

단, 하나의 함수에 모든 것을 해결하려하면 복잡했기 때문에 3개의 함수로 분리하여 요구사항을 코드로 작성합니다.

 

 

 

문제 요구 사항

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. 생성된 문자열을 반환합니다.

 

 

문제 요구사항을 코드로 구현

 

1. 입력이 빈 문자열인 경우, 빈 문자열을 반환합니다.

 

def solution(p):
    return p if correct(p) else recursive(p)
    
def recursive(sentence):
    if sentence == '':
        return '' # 입력 문자열이 공백이면, 공백 return
    u, v = detatch(sentence)
    if correct(u):
        return u + recursive(v)
    else:
        return '(' + recursive(v) + ')' + ''.join(list(map(lambda x: ")" if x =="(" else "(", u[1:-1])))

 

2. 문자열 w를 두 "균형잡힌 괄호 문자열" u, v로 분리합니다. 단, u는 "균형잡힌 괄호 문자열"로 더 이상 분리할 수 없어야 하며, v는 빈 문자열이 될 수 있습니다.

 

- 분리하고자 하는 문자열을 우선 queue로 만듭니다.

- u가 우선 균형잡힌 문자열의 형태를 띄면 곧 바로 return 시켜 u, v를 분리합니다.

def detatch(sentence):
    temp_que = deque(sentence)
    left, right = 0, 0
    u, v = '', ''

    while temp_que:
        u += temp_que.popleft()
        if u[-1] == "(":
            left += 1
        else:
            right += 1

        if left == right:
            break

    v = ''.join(list(temp_que))
    return u, v

 

3. 문자열 u가 "올바른 괄호 문자열" 이라면 문자열 v에 대해 1단계부터 다시 수행합니다.

   3-1. 수행한 결과 문자열을 u에 이어 붙인 후 반환합니다.

 

- correct() 함수에서 True가 리턴될 때와 그렇지 않을 때를 분기처리합니다.

- 문자열 u가 올바른 괄호 문자열 일 때, v 문자열에 대해 다시 recursive() 함수의 인자로 넣어 호출합니다. (재귀 구조)

def correct(sentence):
    stack = []
    for sen in sentence:
        if sen == "(":
            stack.append(sen)
        else:
            if len(stack) == 0:
                return False
            else:
                stack.pop()

    return True if len(stack) == 0 else False
    
def recursive(sentence):
    if sentence == '':
        return ''
    u, v = detach(sentence)
    if correct(u):
        return u + recursive(v)
    else:
        # return '(' + recursive(v) + ')' + reverse(u[1:-1])
        return '(' + recursive(v) + ')' + ''.join(list(map(lambda x: ")" if x =="(" else "(", u[1:-1])))

 

 

4. 문자열 u가 "올바른 괄호 문자열"이 아니라면 아래 과정을 수행합니다.

   4-1. 빈 문자열에 첫 번째 문자로 '('를 붙입니다.

   4-2. 문자열 v에 대해 1단계부터 재귀적으로 수행한 결과 문자열을 이어 붙입니다.

   4-3. ')'를 다시 붙입니다.

   4-4. u의 첫 번째와 마지막 문자를 제거하고, 나머지 문자열의 괄호 방향을 뒤집어서 뒤에 붙입니다.

   4-5. 생성된 문자열을 반환합니다.

 

- 올바른 괄호가 아니므로 else 이하의 코드를 수행합니다.

- 4-4 요구 사항을 주의해야하는데, u[1:-1] 문자열을 그저 뒤집는 것이 아닌, u[1:-1]의 원소 하나하나의 괄호를 뒤집어 줘야합니다.

- 이 사항은 람다식을 이용하여 구현했습니다.

def recursive(sentence):
    if sentence == '':
        return ''
    u, v = detach(sentence)
    if correct(u):
        return u + recursive(v)
    else:
        return '(' + recursive(v) + ')' + ''.join(list(map(lambda x: ")" if x =="(" else "(", u[1:-1])))

 

전체 코드

from collections import deque


def detatch(sentence):
    temp_que = deque(sentence)
    left, right = 0, 0
    u, v = '', ''

    while temp_que:
        u += temp_que.popleft()
        if u[-1] == "(":
            left += 1
        else:
            right += 1

        if left == right:
            break

    v = ''.join(list(temp_que))
    return u, v


def recursive(sentence):
    if sentence == '':
        return ''
    u, v = detatch(sentence)
    if correct(u):
        return u + recursive(v)
    else:
        return '(' + recursive(v) + ')' + ''.join(list(map(lambda x: ")" if x =="(" else "(", u[1:-1])))


def correct(sentence):
    stack = []
    for sen in sentence:
        if sen == "(":
            stack.append(sen)
        else:
            if len(stack) == 0:
                return False
            else:
                stack.pop()

    return True if len(stack) == 0 else False


def solution(p):
    return p if correct(p) else recursive(p)