[Programmers] 괄호 변환2

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

이 문제는

"너는 시키는대로 얼마나 잘해내니?"

이걸 물어본 것 같은 문제였다.

나는 세개의 함수로 구현하였다.

1. 올바른 괄호 문자열인지 return 해주는 함수 (right)
2. u와 v로 나누어주는 함수 (devided_uv)
3. 문제에서 나온 그대로 구현한 함수 (solution)

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

차근차근 풀어보려고 기능을 나누어 구현하였는데,
이 코드는 올바른 문자열인지 확인해주는 코드를 그대로 가져왔다.

스택을 이용하였으며

• '(' 괄호열이 몇 개 들어왔는지는 신경쓰지 않고, ')'괄호열이 들어왔을 때를 기준으로 이 문자열이 올바른 괄호 문자열인지 체크해주었다.

• ')'괄호열이 들어왔을 때 스택 안에 값이 있다면 무조건 '(' 문자열 이기 때문에 pop을 해줘서 '('을 빼준다.

• 값이 없는데 ')'이 들어왔다면 올바르지 않은 문자열이기 때문에 False를 return해준다.

• 또한 스택에 값이 남아있다면 이 또한 올바르지 않은 문자열이기 때문에 False를 return해준다.

• 나머지는 True를 리턴해주었다.

tmi)
원래 여기서 False가 return되면 그 때 올바른 괄호열로 바꾸어주는 솔루션을 만들고자 했는데,
이 때 내가 구상한 알고리즘으로 혼자서 구현해보고자 했다.
여러차례 시도 후 더 이상 아이디어가 떠오르지 않아서 다시 문제를 읽어 봤더니,
있는 그대로 구현하면 올바른 문자열이 되는 것이었다..! 역시 문제를 잘 읽어야 혀....

# u와 v를 균형잡힌 문자열로 변환하기
# (, )의 갯수가 똑같으면 균형잡힌 문자열이므로 같아지는 때에 u,v로 분리한다.

def devide_uv(s):
    count_i = 0 # ( 의 갯수
    count_j = 0 # ) 의 갯수

    for i in range(len(s)):
        if s[i] == '(':
            count_i+=1
        else:
            count_j+=1
        if count_i == count_j:
            break
    return s[:i+1], s[i+1:]

이 부분은 u와 v로 나눠주는 기능인데,

'('와 ')'의 갯수를 카운트 해 준 다음, 두 갯수가 맞았을 때 균형잡힌 문자열이 되므로 이 때 u와 v로 나누어준다.

def solution(p):
    # 1번
    if p == '':
        return ''

    # 2번
    u, v = devide_uv(p)

    # 3번
    if right(u):
        # 3-1번
        return u + solution(v)
    else:
        # 4-1,2,3번
        answer = '('
        answer += solution(v)
        answer += ')'
        
        # 4-4번
        temp_u = u[1:-1]
        for i in temp_u:
            if i == '(':
                answer += ')'
            else:
                answer += '('

        return answer

마지막으로 문제에 있는 설명을 그대로 구현하였다.

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

마지막 부분을 구현하면서 어...? 이래도 되나...? 싶었는데 되더군....

요청하는 대로 얼마나 잘 구현해내는 사람인지 확인하는 문제였다는 생각이 든다.