[프로그래밍문제] 중괄호 해제하기

 

def recur(string):
    # 숫자찾기
    a = -1
    for i in range(len(string)):
        if ord('0')<= ord(string[i]) <=ord('9'):
            a = int(string[i])
            break
    if a == -1:
        return string    
 
    # 최종 괄호 찾기
    start = 0
    end = 0
    count = 0
    for i in range(len(string)):
        if string[i]=="(":
            count +=1
            if count ==1:
                start =i
        if string[i]==")":
            count -=1 
            if count ==0:
                end = i
                break
 
    ret = string[start+1: end]*a
    if start>1:
        ret = string[:start-1]+ ret    
    if end < len(string)-1:
        ret = ret + string[end+1:] 
    return recur(ret)
 
 
 
A = ["3(mj2(bj)3(hi))", "mjbjbjhihihimjbjbjhihihimjbjbjhihihi"]
B = ["5(a)2(b)1(c)", "aaaaabbc"]
C = ["2(a2(b)3(c))", "abbcccabbccc"]
 
print(recur(A[0]) == A[1])
print(recur(B[0]) == B[1])
print(recur(C[0]) == C[1])