programmers.co.kr/learn/courses/30/lessons/60058
코딩테스트 연습 - 괄호 변환
카카오에 신입 개발자로 입사한 콘은 선배 개발자로부터 개발역량 강화를 위해 다른 개발자가 작성한 소스 코드를 분석하여 문제점을 발견하고 수정하라는 업무 과제를 받았습니다. 소스를 컴
programmers.co.kr
문제 설명
카카오에 신입 개발자로 입사한 콘은 선배 개발자로부터 개발역량 강화를 위해 다른 개발자가 작성한 소스 코드를 분석하여 문제점을 발견하고 수정하라는 업무 과제를 받았습니다. 소스를 컴파일하여 로그를 보니 대부분 소스 코드 내 작성된 괄호가 개수는 맞지만 짝이 맞지 않은 형태로 작성되어 오류가 나는 것을 알게 되었습니다.
수정해야 할 소스 파일이 너무 많아서 고민하던 콘은 소스 코드에 작성된 모든 괄호를 뽑아서 올바른 순서대로 배치된 괄호 문자열을 알려주는 프로그램을 다음과 같이 개발하려고 합니다.
용어의 정의
'(' 와 ')' 로만 이루어진 문자열이 있을 경우, '(' 의 개수와 ')' 의 개수가 같다면 이를 균형잡힌 괄호 문자열이라고 부릅니다.
그리고 여기에 '('와 ')'의 괄호의 짝도 모두 맞을 경우에는 이를 올바른 괄호 문자열이라고 부릅니다.
예를 들어, "(()))("와 같은 문자열은 균형잡힌 괄호 문자열 이지만 올바른 괄호 문자열은 아닙니다.
반면에 "(())()"와 같은 문자열은 균형잡힌 괄호 문자열 이면서 동시에 올바른 괄호 문자열 입니다.
'(' 와 ')' 로만 이루어진 문자열 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의 앞뒤 문자를 제거하고, 나머지 문자의 괄호 방향을 뒤집으면 "()"이 됩니다.
- 따라서 생성되는 문자열은 "(" + "()" + ")" + "()"이며, 최종적으로 "(())()"를 반환합니다.
- 처음에 그대로 둔 문자열에 반환된 문자열을 이어 붙이면 "()" + "(())()" = "()(())()"가 됩니다.
☆ 나의 풀이
# 실패
def solution(p):
u, v = '', ''
def _split(u,v):
cnt=0
for i, j in enumerate(p):
if j=="(":
cnt+=1
u+="("
else:
cnt-=1
u+=")"
if cnt<0:
v=p[i:] # 불완전
u=p[:i] # 완전
return u, v
return u, v
def _del(u, v):
if len(u)!=0:
u=u[1:-1]
def reverse(u, v):
v=list(v)
v.reverse()
print(v)
print(u)
u=u+str(v)
v=''
#print(u)
return u, v
while True:
u,v=_split(u,v)
if len(v)==0: # 완전
return u
#u, v=reverse(u, v)
#if len(v)==0:
# return str(u)
#print(u,v)
조건이 많아서 풀기 어려웠던 듯.. deque 사용해야겠다는 생각을 못했다 ㅠ
is_correct와 detatch를 합쳐서 만들었었다 뒤집기는 list로 변환 후 reverse함수 사용한 후 다시 문자열로 바꿨는데 넘 비효율적인 듯. recursive함수 없이 나는 solution함수에서 while문 돌렸다. u와 v로 분리한 후에 올바른 괄호인지 확인한다.
# 성공
def _split(p): # 문자열p를 u, v로 분리하는 함수
u, v='',''
cnt=0
for i, j in enumerate(p):
if j=="(":
cnt+=1
else:
cnt-=1
if cnt==0:
v=p[i+1:] # 불완전
u=p[:i+1] # 완전
return u, v
def check(u): # 균형잡힌 괄호 문자열이면 True반환
cnt=0
for i in u:
if i=="(":
cnt+=1
else:
cnt-=1
if cnt<0: # cnt가 음수라는 것은 균형이 맞지 않다는 것
return False
return True
def reverse(u): # 문자열 u를 뒤집는 함수
return ''.join([')' if i=='(' else '(' for i in u])
def solution(p): # 재귀 사용
if p=='': # 1단계
return ''
u, v=_split(p) # 2단계 : 분리
if check(u): # 균형잡힌 괄호 문자열인지 검사
return u+solution(v) # 3단계와 3-1 : u가 올바른 괄호 문자열일 때
else:
return '('+solution(v)+')'+reverse(u[1:-1]) # 4단계
문제에 제시된 구현 순서에 따라 만든다. 디테일한 조건을 잘 놓친다.ㅠㅠ
1. solution함수에서 재귀 오류났는데 알고보니 solution(p)를 넣어서 무한반복이었다 😂
2. check함수에서 왜 죄다 True가 나오나 했는데 return에 False 조건을 넣어주면 )(의 경우 루프 다 돌면 0이라서 True가 반환되기에 잘못된 것이다.
3. list conprehension 아직 익숙하지 않구나 reverse 함수 만들면서 for문 쓰기 싫어서 꼼수 썼던ㅋ
4. _split함수에서 u랑 v에 값 넣을 때 슬라이싱 값을 잘못했다 cnt==0일 때 p[i]까지가 균형이 맞는 문자열이다. i를 포함하기 위해 i+1
4단계에서 solution(v)가 더 앞에 나오는 이유?
ex) p='()))((()' result='()(()))()'
u='()' # 완전
v='))((()' # 불완전
return u+solution(v)
# v는 solution에 넣어서
u='))(('
v='()'
return '('+solution(v)+')'+reverse(u[1:-1])
# solution(v) 실행
u='()'
v=''
return u+solution(v)
# solution(v) 실행하는데 v==''"이므로 return ''"
return '(())'+reverse(u[1:-1])
# reverse(u[1:-1]) 실행
()
return '(())()'
# return u+solution(v)의 결과가 리턴됨 '()(())()'
👉 결론 : 재귀를 돌리면 v속에서 u를 찾는 것이므로 solution(v)가 앞에 온다.
☆ 다른 사람 풀이
from collections import deque
def is_correct(string): # 올바른 괄호인지 확인
stack = []
for s in string:
if s == '(':
stack.append(s)
elif stack:
stack.pop()
return not stack
def detatch(string): # u, v로 분리
str_que = deque(string)
left, right = 0, 0
u, v = "", ""
while str_que: # 큐사용
u += str_que.popleft()
if u[-1] == '(':
left += 1
else:
right += 1
if left == right:
break # 균형 잡힌 괄호
v = ''.join(list(str_que))
return u, v
def reverse(u): # 뒤집기
return ''.join([')' if s == '(' else '('for s in u]) # 반복하면서 괄호를 바꾸고 문자열로 만듦
def recursive(string):
if string == '': # 1번
return ''
u, v = detatch(string) # 2번
if is_correct(u): # 3번
return u + recursive(v)
else: # 4번
return '(' + recursive(v) + ')' + reverse(u[1:-1])
def solution(p):
return p if is_correct(p) else recursive(p)
앞뒤에서 제거할 수 있게 deque를 사용.
is_correct함수에서 반환할 때 stack이 값이 있으면 False고 not으로 인해 True가 반환됨. solution함수에서 p를 그대로 반환한다. else의 경우에 recursive함수를 호출한다. detatch함수를 호출하여 left와 right를 이용하여 균형 잡힌 괄호인지 판단한다. 문자열로 v를 반환한다. 다시 is_correct함수를 호출한다. else는 u에서 처음과 끝을 제외한 문자열을 reverse함수에 넣어서 뒤집는다.
# u, v로 분리하는 함수
def divide(w):
openP=0
closeP=0
for i in range(len(w)):
if w[i]=='(':
openP+=1
else:
closeP+=1
if openP==closeP:
return w[:i+1], w[i+1:]
#문자열 u가 올바른 괄호 문자열인지 확인
def isBalanced(u):
stack=[]
for p in u:
if p=='(':
stack.append(p)
else:
if not stack: #스택이 비었다면
return False
stack.pop()
return True
def solution(w):
#과정1 아무것도 없으면
if not w:
return ""
#과정2 분리
u, v=divide(w)
#과정3 올바른 문자열인지
if isBalanced(u):
#과정3-1
return u+solution(v)
#과정4
else: # 올바른 괄호 문자열이 아니라면
#과정4-1
answer='('
#과정4-2
answer+=solution(v)
#과정4-3
answer+=')'
#과정4-4
for p in u[1:len(u)-1]:
if p=='(':
answer+=')'
else:
answer+='('
return answer #과정 4-5
이 코드는 문자열을 분리하는 함수, 올바른 괄호인지 판단하는 함수로 구성됨. 뒤집는 함수는 따로 만들지 않고 solution함수 내에서 for문으로 해결하고 있음.
def solution(p):
if p=='': return p
r=True; c=0 # r: 올바른 괄호 문자열이면 True, c:'('와 ')'를 count하는 역할
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:]) # p[:i+1]이 u, p[i+1:]이 v
else:
return '('+solution(p[i+1:])+')'+''.join(list(map(lambda x:'(' if x==')' else ')',p[1:i]) ))
하나의 함수내에서 다 해결하는 😮
올바른 괄호 문자열인지 판단하는 알고리즘이 나랑 같다.
u랑 v로 나누지 않고 슬라이싱으로 해결.
join함수에 map을 사용해서 문자열을 거꾸로 뒤집어 주고있다.
'coding test' 카테고리의 다른 글
| [파이썬] 소수 찾기 (0) | 2020.12.16 |
|---|---|
| [파이썬] 다음 큰 숫자 (0) | 2020.12.14 |
| [파이썬] 나누어 떨어지는 숫자 배열 (0) | 2020.12.09 |
| [파이썬] 올바른 괄호 (0) | 2020.12.09 |
| [파이썬] 가장 큰 정사각형 찾기 (0) | 2020.12.08 |