[프로그래머스] 올바른 괄호 문자열 만들기 (2020 KAKAO BLIND) / python

1️⃣ 문제 설명

코딩테스트 연습 - 괄호 변환

"균형잡힌 괄호 문자열" p가 매개변수로 주어질 때,

주어진 알고리즘을 수행해 "올바른 괄호 문자열"로 변환한 결과를 return

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

• 문자열 p를 이루는 '(' 와 ')' 의 개수는 항상 같습니다.
• 만약 p가 이미 "올바른 괄호 문자열"이라면 그대로 return 하면 됩니다.

2️⃣ 풀이 전 계획과 생각

구현 문제다.

올바른 괄호 체크는 스택으로 체크해주자.

입력은 무조건 균형잡힌 괄호 문자열로 주어지기에,

입력을 올바른 괄호 문자열 바꿔주는 로직을 거친 후에, return 해주면 된다.

• 아래의 로직을 그대로 구현한다.
  • 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. 생성된 문자열을 반환합니다.
• 올바른 괄호 문자열인지 체크해 주는 함수
  • ()() 짝이 맞아야 하므로, ) 앞의 값이 (가 있어야 한다. ( 가 없다면, 올바른 괄호 문자열이 아니다.
  • p가 (라면, 스택에 넣어준다.
  • P가 )라면, 스택을 빼준다. 왜? ()맞춰주기 위해서다.
  • → 빼줄 값이 없다면, 균형잡힌 괄호 문자열이 아니다.
• 문자열 u, v로 분리해주는 함수
  • "균형잡힌 괄호 문자열"로 더 이상 분리할 수 없어야 하며,
    • → 입력값의 p가 균형 잡힌 괄호 문자열이 되면, 잘라준다.
    • 그렇다면, p를 순회해서 왼쪽 괄호 갯수, 오른쪽 괄호 갯수가 같을 때, 분리해준다.
    • 그러면, 나머지도 균형 잡힌 괄호 문자열이 되기 때문이다.

3️⃣ 풀이

def is_right(u):
    stack = []
    
    for p in u:
        if p == '(':
            stack.append(p)
        else:
            if not stack:   return False
            stack.pop()
            
    return True

def divide_two_blanced(p):
    left_count =  0
    right_count = 0
    
    for i in range(len(p)):
        if p[i] == '(': left_count += 1
        else :  right_count += 1
        if left_count == right_count:   return p[:i+1], p[i+1:]

def solution(p):
    # 1.  입력이 빈 문자열인 경우, 빈 문자열을 반환합니다.
    if len(p) == 0:
        return ""
    # 2. 문자열 w를 두 "균형잡힌 괄호 문자열" u, v로 분리합니다.
    u, v = divide_two_blanced(p)
    
    # 3. 문자열 u가 "올바른 괄호 문자열" 이라면 문자열 v에 대해 1단계부터 다시 수행합니다.
    if is_right(u):
        # 3-1. 수행한 결과 문자열을 u에 이어 붙인 후 반환합니다.
        return u + solution(v)
    # 4. 문자열 u가 "올바른 괄호 문자열"이 아니라면 아래 과정을 수행합니다.
    else:
        # 4-1. 빈 문자열에 첫 번째 문자로 '('를 붙입니다.
        answer = '('
        # 4-2. 문자열 v에 대해 1단계부터 재귀적으로 수행한 결과 문자열을 이어 붙입니다.
        answer += solution(v)
        # 4-3. ')'를 다시 붙입니다.
        answer += ')'
        # 4-4. u의 첫 번째와 마지막 문자를 제거하고, 나머지 문자열의 괄호 방향을 뒤집어서 뒤에 붙입니다.
        u = u[1:-1]
        for element in u:
            if element == ')':
                answer += '('
            else:
                answer += ')'
         # 4-5. 생성된 문자열을 반환합니다.
        return answer

4️⃣ 실수 및 알게된 사실

• -i에서 맨끝의 값까지 가져오고 싶으면 list[-i:] 해주면된다.
  나는 순간 list[-i:-1]로 해주었다.
• str[len(str):] 의 경우는 index 초과 에러가 안난다. 빈문자열을 리턴해준다.