[ Programmers / CodingTest / Python ] 괄호 변환

황승환·2022년 3월 24일
0

Python

목록 보기
255/498

문제 설명

카카오에 신입 개발자로 입사한 "콘"은 선배 개발자로부터 개발역량 강화를 위해 다른 개발자가 작성한 소스 코드를 분석하여 문제점을 발견하고 수정하라는 업무 과제를 받았습니다. 소스를 컴파일하여 로그를 보니 대부분 소스 코드 내 작성된 괄호가 개수는 맞지만 짝이 맞지 않은 형태로 작성되어 오류가 나는 것을 알게 되었습니다.
수정해야 할 소스 파일이 너무 많아서 고민하던 "콘"은 소스 코드에 작성된 모든 괄호를 뽑아서 올바른 순서대로 배치된 괄호 문자열을 알려주는 프로그램을 다음과 같이 개발하려고 합니다.

용어의 정의

'(' 와 ')' 로만 이루어진 문자열이 있을 경우, '(' 의 개수와 ')' 의 개수가 같다면 이를 균형잡힌 괄호 문자열이라고 부릅니다.
그리고 여기에 '('와 ')'의 괄호의 짝도 모두 맞을 경우에는 이를 올바른 괄호 문자열이라고 부릅니다.
예를 들어, "(()))("와 같은 문자열은 "균형잡힌 괄호 문자열" 이지만 "올바른 괄호 문자열"은 아닙니다.
반면에 "(())()"와 같은 문자열은 "균형잡힌 괄호 문자열" 이면서 동시에 "올바른 괄호 문자열" 입니다.

'(' 와 ')' 로만 이루어진 문자열 w가 "균형잡힌 괄호 문자열" 이라면 다음과 같은 과정을 통해 "올바른 괄호 문자열"로 변환할 수 있습니다.

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가 매개변수로 주어질 때, 주어진 알고리즘을 수행해 "올바른 괄호 문자열"로 변환한 결과를 return 하도록 solution 함수를 완성해 주세요.

매개변수 설명

  • p는 '(' 와 ')' 로만 이루어진 문자열이며 길이는 2 이상 1,000 이하인 짝수입니다.
  • 문자열 p를 이루는 '(' 와 ')' 의 개수는 항상 같습니다.
  • 만약 p가 이미 "올바른 괄호 문자열"이라면 그대로 return 하면 됩니다.

입출력 예

p			result
"(()())()"	"(()())()"
")("		"()"
"()))((()"	"()(())()"

입출력 예에 대한 설명

입출력 예 #1
이미 "올바른 괄호 문자열" 입니다.

입출력 예 #2

  • 두 문자열 u, v로 분리합니다.
    - u = ")("
    - v = ""
  • u가 "올바른 괄호 문자열"이 아니므로 다음과 같이 새로운 문자열을 만듭니다.
    - v에 대해 1단계부터 재귀적으로 수행하면 빈 문자열이 반환됩니다.
    - u의 앞뒤 문자를 제거하고, 나머지 문자의 괄호 방향을 뒤집으면 ""이 됩니다.
    - 따라서 생성되는 문자열은 "(" + "" + ")" + ""이며, 최종적으로 "()"로 변환됩니다.

입출력 예 #3

  • 두 문자열 u, v로 분리합니다.
    - u = "()"
    - v = "))((()"
  • 문자열 u가 "올바른 괄호 문자열"이므로 그대로 두고, v에 대해 재귀적으로 수행합니다.
  • 다시 두 문자열 u, v로 분리합니다.
    - u = "))(("
    - v = "()"
  • u가 "올바른 괄호 문자열"이 아니므로 다음과 같이 새로운 문자열을 만듭니다.
    - v에 대해 1단계부터 재귀적으로 수행하면 "()"이 반환됩니다.
    - u의 앞뒤 문자를 제거하고, 나머지 문자의 괄호 방향을 뒤집으면 "()"이 됩니다.
    - 따라서 생성되는 문자열은 "(" + "()" + ")" + "()"이며, 최종적으로 "(())()"를 반환합니다.
  • 처음에 그대로 둔 문자열에 반환된 문자열을 이어 붙이면 "()" + "(())()" = "()(())()"가 됩니다.

접근 방법

이번 문제는 구현 문제로, 주어진 조건에 맞게 재귀 함수로 작성하였다. 처음에는 문제 이해가 되지 않아 시간이 조금 걸렸지만, 한 단계씩 차근차근 함수로 구현하여 해결할 수 있었다.

다음과 같은 함수들을 작성하여 풀이하였다.

  • 문자열을 조건에 맞게 자르기 위한 인덱스를 반환하는 cutting 함수
  • 균형잡힌 괄호 문자열인지 확인하기 위한 counting 함수
  • 올바른 괄호 문자열인지 확인하기 위한 checking 함수
  • 괄호 문자열의 괄호 방향을 뒤집는 opposite 함수
  • 괄호 문자열을 최종 문자열로 만들어 반환하는 change 함수

이 함수들을 적절하게 사용하여 문제를 해결하였다.

  • 문자열을 자를 인덱스를 반환하는 cutting 함수를 s를 인자로 갖도록 선언한다.
    -> '('와 ')'의 카운팅 변수 cnt1, cnt2를 0으로 선언한다.
    -> s의 길이만큼 반복하는 i에 대한 for문을 돌린다.
    --> 만약 s[i]가 '('일 경우, cnt1을 증가시킨다.
    --> 그 외의 경우 cnt2를 증가시킨다.
    --> 만약 cnt1과 cnt2가 같을 경우, i를 반환한다.
    -> s의 길이-1을 반환한다.
  • 균형잡힌 괄호 문자열인지 확인하는 counting 함수를 s를 인자로 갖도록 선언한다.
    -> '('와 ')'의 카운팅 변수 cnt1, cnt2를 0으로 선언한다.
    -> s의 길이만큼 반복하는 i에 대한 for문을 돌린다.
    --> 만약 s[i]가 '('일 경우, cnt1을 증가시킨다.
    --> 그 외의 경우 cnt2를 증가시킨다.
    -> 만약 cnt1과 cnt2가 같을 경우, True를 반환한다.
    -> 그 외의 경우, False를 반환한다.
  • 올바른 괄호 문자열인지 확인하는 checking함수를 s를 인자로 갖도록 선언한다.
    -> 스택으로 사용할 리스트 stack을 선언한다.
    -> s의 길이만큼 반복하는 i에 대한 for문을 돌린다.
    --> 만약 s[i]가 '('일 경우,
    ---> stack에 s[i]를 넣는다.
    --> 그 외의 경우,
    ---> 만약 stack이 비어있을 경우,
    ----> False를 반환한다.
    ---> stack을 pop()한다.
    -> True를 반환한다.
  • 괄호를 반대로 돌리는 opposite함수를 s를 인자로 갖도록 선언한다.
    -> s를 리스트로 변환한다.
    -> s의 길이만큼 반복하는 i에 대한 for문을 돌린다.
    --> 만약 s[i]가 '('일 경우,
    ---> s[i]를 ')'로 갱신한다.
    --> 그 외의 경우,
    ---> s[i]를 '('로 갱신한다.
    -> s를 문자열로 반환한다.
  • 올바른 괄호 문자열을 반환하는 함수 change를 s를 인자로 갖도록 선언한다.
    -> 만약 s가 빈 문자열일 경우, 빈 문자열을 반환한다.
    -> s를 자를 인덱스를 저장할 변수 idx에 cutting(s)의 반환값을 저장한다.
    -> 자른 두 문자열 u, v를 s[:idx+1]s[idx+1:]로 선언한다.
    -> 만약 checking(u)가 True일 경우,
    --> u+change(v)를 반환한다.
    -> 만약 checking(u)가 False일 경우,
    --> 임시 변수 tmp를 '('+change(v)+')'+opposite(u[1:-1])로 선언한다.
    --> tmp를 반환한다.
  • 정답을 저장할 변수 answer를 change(p)로 선언한다.
  • answer를 반환한다.

solution.py

def solution(p):
    def cutting(s):
        cnt1, cnt2=0, 0
        for i in range(len(s)):
            if s[i]=='(':
                cnt1+=1
            else:
                cnt2+=1
            if cnt1==cnt2:
                return i
        return len(s)-1
    def counting(s):
        cnt1, cnt2=0, 0
        for i in range(len(s)):
            if s[i]=='(':
                cnt1+=1
            else:
                cnt2+=1
        if cnt1==cnt2:
            return True
        else:
            return False
    def checking(s):
        stack=[]
        for i in range(len(s)):
            if s[i]=='(':
                stack.append(s[i])
            else:
                if not stack:
                    return False
                stack.pop()
        return True
    def opposite(s):
        s=list(s)
        for i in range(len(s)):
            if s[i]=='(':
                s[i]=')'
            else:
                s[i]='('
        return ''.join(s)
    def change(s):
        if not s:
            return ''
        idx=cutting(s)
        u, v=s[:idx+1], s[idx+1:]
        if checking(u):
            return u+change(v)
        if not checking(u):
            tmp='('+change(v)+')'+opposite(u[1:-1])
            return tmp
    answer = change(p)
    return answer

profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글