프로그래머스 / 코딩 테스트 / 괄호 변환

문제 설명

 

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

 

문제 해결

 

일단 용어부터 확인해야한다.

'(', ')' 문자의 개수가 같다면 '균형잡힌 괄호 문자열'이다.

소괄호의 짝이 모두 맞다면 '올바른 괄호 문자열'이다.

 

일단 균형잡힌 괄호 문자열을 구분할 때는 문자의 개수를 count하는 형태로 구현하면 된다.

균형 잡힌 문자열은 2번 과정에서 밖에 사용하지 않아서 solve 함수와 합쳐서 구현했다.

아래는 균형잡힌 문자열을 u, v에 나눠담는 코드이다.

 

    u = None
    v = ''
    op = 0
    cp = 0
    for i in range(len(w)):
        if w[i] == '(':
            op +=1
        else:
            cp +=1
        if op == cp:
            u = w[:i+1]
            v = w[i+1:]
            break

 

올바른 괄호 문자열을 어떻게 구분할까 고민했고 stack의 후입선출 구조를 이용했다.

 

def isCorrect(s):
    stack = []
    
    for c in s:
        if c == '(':
            stack.append(c)
        else:
            if len(stack) == 0:
                return False
            stack.pop()
    
    return True

 

문제 설명에서 '재귀'라는 단어가 나온 것을 보니 재귀함수를 구현해야 풀 수 있는 문제일 확률이 높다.

문제에서 제시한 알고리즘이 비교적 복잡하지만 한줄씩 구현하라는 데로 구현하면 풀 수 있다.

 

def isCorrect(s):
    stack = []
    
    for c in s:
        if c == '(':
            stack.append(c)
        else:
            if len(stack) == 0:
                return False
            stack.pop()
    
    return True


def solve(w):
    # 1
    if len(w) == 0:
        return ""
    
    # 2
    u = None
    v = ''
    op = 0
    cp = 0
    for i in range(len(w)):
        if w[i] == '(':
            op +=1
        else:
            cp +=1
        if op == cp:
            u = w[:i+1]
            v = w[i+1:]
            break
    
    # 3
    if isCorrect(u):
        #3-1
        return u + solve(v)
    
    #4
    else:
        #4-1
        t = '('
        #4-2
        t += solve(v)
        #4-3
        t += ')'
        #4-4
        t += "".join([')' if i == '(' else '(' for i in u[1:len(u)-1]])
        #4-5
        return t
        


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

 

재귀함수를 응용해본 경험이 상당히 적어서 3, 4번 문장을 어떻게 코드로 구현할지 상당히 고민했다.