[코딩 테스트] 프로그래머스 - 괄호 변환 (2020 Kakao) in Python

'('와 ')'로만 이루어진 문자열 s에서, 괄호들을 올바르게 배치해서 반환할 것.

 

이 문제는 문제 자체에서 알고리즘을 이미 나열해놨다.

이에 맞게 구현만 하면 되는 문제.

재귀적인 성질을 갖고 있다.

 

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

 

팁 - 구현문제는, 실수를 최대한 줄이기 위해서 메소드를 나눠주는 게 좋다고 한다.

따라서, '균형잡힌 괄호 문자열'에 관련된 메소드, '올바른 괄호 문자열'에 관련된 메소드를 각각 만들자.

 

- Step 2에서 균형잡힌 괄호 문자열 두 개로 분리하기 위해, 균형잡힌 괄호 문자열의 인덱스를 반환하는 메소드를 만들 것.

- Step 3에서 쓸 수 있도록 올바른 괄호 문자열인지 확인하는 메소드 만들 것.

 

 

코드 구현

 

'''
Parentheses Test
programmers.co.kr/learn/courses/30/lessons/60058

Fix the parenthese in the given string s
03/08/21
'''

# 균형잡힌 괄호 문자열의 인덱스 반환
def balanced_index(p):
    count = 0 # 왼쪽 괄호의 개수
    for i in range(len(p)):
        if p[i] == '(':
            count += 1
        else:
            count -= 1
        if count == 0:
            return i

# 올바른 괄호 문자열인 지 확인
def check_proper(p):
    count = 0 # 왼쪽 괄호의 개수
    for i in p:
        if i == '(':
            count += 1
        else:
            if count == 0:
                return False
            else:
                count -= 1

    return True

# solution 함수
def solution(p):
    answer = ''
    # 1. if empty, return empty string
    if p == '':
        return answer
    # 2. split into two 'balanced parentheses string'
    index = balanced_index(p)
    u = p[:index+1]
    v = p[index+1:]
    # 3. if u is 'proper p str' apply Step 1~ to v
    if check_proper(u):
        answer = u + solution(v)
    
    # 4. if not,
    # 4-1. add '(' to answer
    else:
        answer = '('
        # 4-2. apply Step 1~ to v
        answer += solution(v)
        # 4-3. add ')' to answer
        answer += ')'
        # 4-4. remove the first and last letters of u,
        # and add the flipped parentheses of the remainig string
        for j in u[1:-1]:
            if j == '(':
                answer += ')'
            else:
                answer += '('
    return answer