[프로그래머스] 괄호 변환 Python Code

 

프로그래머스 괄호 변환을 파이썬으로 풀어보겠다.

이 문제는 2020년 카카오 코딩 테스트 문제고

레벨은 2로 되어있다.

---

문제

 

이 문제의 설명은 길지만 용어의 정의를 차근차근 코드로 옮겨 적으면 된다.

매개변수 설명에 양쪽 괄호의 개수가 항상 같으므로

예외사항을 따로 고려하지 않아도 된다.

---

구현

2번에 "균형 잡힌 문자열" u, v로 구분하는 함수,

3, 4번을 나누는 기준이 되는 "올바른 괄호 문자열" 판단 함수를 분리해서 구현했다.

 

문제 설명에 구현이 다 나와있어서 별로 생각할 게 없는 문제다.

---

코드

def divUV(p):
    l, r = 0,0
    for i in range(len(p)):
        if p[i] == '(':
            l +=1
        else:
            r +=1
        if l==r:
            return p[:i+1], p[i+1:]

def isCorrect(p):
    temp=[]
    for c in p:
        if c== '(':
            temp.append(c)
        else:
            if not temp:
                return False
            temp.pop()
    return False if temp else True

def solution(p):
    if p == "":
        return ""
    u,v = divUV(p)

    if isCorrect(u):
        return u + solution(v)
    else:
        temp="("+solution(v)+")"

        for c in u[1:-1]:
            if c == '(':
                temp += ')'
            else:
                temp+='('

    return temp

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

이 과정을 그대로 구현했다.

1단계부터 다시 수행하는 부분은 재귀 호출로 구현했다.

 

문자열을 나누는 함수는 divUV고,

올바른 문자열인지 판단하는 함수는 isCorrect다.

 

위 설명과 순서도 같기 때문에 참고하면서 코드를 보면 좋을 것 같다.

---

다른 풀이

 

짧은 코드를 가져와 봤다.

def solution(p):
    if p=='': 
    	return p
    
    r=True
    c=0
    
    for i in range(len(p)):
        if p[i]=='(': 
        	c-=1
        else: 
        	c+=1
        if c>0: 
        	r=False
        if c==0:
            if r:
                return p[:i+1]+solution(p[i+1:])
            else:
                return '('+solution(p[i+1:])+')'+''.join(list(map(lambda x:'(' if x==')' else ')',p[1:i]) ))

이 코드는 나누는 것과 올바른 문자열인지 판단하는 함수를 한 번에 적용했다.

 

c를 통해 0이면 균형 잡힌 문자열이고

r이 True면 올바른 문자열임을 확인할 수 있도록 구현되었다.

map, lambda를 통해 '('와 ')'를 한 라인에서 바꾼 것이

파이써닉한 코드인 듯하다.

 

이 정도면 가독성도 나쁘지 않기 때문에 참고용으로 가져와보았다.