[Swift] 프로그래머스 - 카카오 괄호 변환

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

코딩테스트 연습 - 괄호 변환

 

 

함수 설명

1. splitUV 함수로 균형잡힌 괄호 문자열인 u와 나머지 문자열인 v를 반환합니다.

2. isCorrect 함수로 스택이 아닌 count 변수로 올바른 괄호 문자열인지 반환합니다.

 

'문자열 v에 대해 1단계부터 다시 수행' 이 부분이 solution 함수를 재귀적으로 호출하는 것임을 알면 풀 수 있는 문제입니다.

 

import Foundation

func splitUV(from p: String) -> [String] {
    let p = Array(p)
    var count = 0
    
    for i in 0..<p.count {
        if p[i] == "(" {
            count += 1
        } else {
            count -= 1
        }
        
        if count == 0 {
            return [String(p[0..<i + 1]), String(p[i + 1..<p.count])]
        }
    }
    
    return []
}

func isCorrect(str: String) -> Bool {
    var count = 0
    
    for s in str {
        if s == "(" {
            count += 1
        } else {
            count -= 1
            
            if count < 0 {
                return false
            }
        }
    }
    
    return count == 0
}

func solution(_ p:String) -> String {
//    1. 입력이 빈 문자열인 경우, 빈 문자열을 반환합니다.
    if p == "" { return "" }
    
//    2. 문자열 w를 두 "균형잡힌 괄호 문자열" u, v로 분리합니다. 단, u는 "균형잡힌 괄호 문자열"로 더 이상 분리할 수 없어야 하며, v는 빈 문자열이 될 수 있습니다.
    
    let uv = splitUV(from: p)
    let u = uv[0]
    let v = uv[1]
    
//    3. 문자열 u가 "올바른 괄호 문자열" 이라면 문자열 v에 대해 1단계부터 다시 수행합니다.
//      3-1. 수행한 결과 문자열을 u에 이어 붙인 후 반환합니다.
    if isCorrect(str: u) {
        return u + solution(v)
    } else {
//    4. 문자열 u가 "올바른 괄호 문자열"이 아니라면 아래 과정을 수행합니다.
//      4-1. 빈 문자열에 첫 번째 문자로 '('를 붙입니다.
//      4-2. 문자열 v에 대해 1단계부터 재귀적으로 수행한 결과 문자열을 이어 붙입니다.
//      4-3. ')'를 다시 붙입니다.
//      4-4. u의 첫 번째와 마지막 문자를 제거하고, 나머지 문자열의 괄호 방향을 뒤집어서 뒤에 붙입니다.
//      4-5. 생성된 문자열을 반환합니다.
        
        var str = "("
        str += solution(v)
        str += ")"
        str += Array(u)[1..<u.count - 1].map { $0 == "(" ? ")" : "(" }
        return str
    }
}