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

문제

프로그래머스

입력

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

 

크게 어려운 건 없었다 먼저 균형잡힌 괄호 문자열과 올바른 괄호 문자열 판단 함수를 만들고 

위 과정에서 분기를 그대로 따라가며 문자열 처리를 하면된다.

 

코드

import Foundation

extension String {
    subscript(_ index: Int) -> String {
        return String(self[self.index(self.startIndex, offsetBy: index)])
    }
}

func isBalanced(_ w: String) -> Bool {
    
    // 균형잡힌 괄호 여부 판단
    if w.isEmpty { return false }
    return w.filter{$0 == "("}.count == w.filter{$0 == ")"}.count
}

func isCorrected(_ w: String) -> Bool {
    
    // 올바른 문자열 판단 여부 
    var stack: [String] = []
    
    for i in 0..<w.count {
        if w[i] == "(" {
            stack.append(w[i])
        } else {
            if stack.isEmpty {
                return false 
            }
            let _ = stack.popLast()
        }
    }
    
    return stack.isEmpty
}

func reversed(_ u: String) -> String {
    
    // 앞뒤 제거 후 뒤집기
    
    var u = u
    let _ = u.removeFirst()
    let _ = u.removeLast()
    
    return u.map({ $0 == ")" ? "(" : ")" }).joined()
}

func solution(_ w:String) -> String {
    if w.isEmpty || isCorrected(w)  {
        return w
    }
    
    var w = w 
    var u = ""
    var v = ""
    
    while !isBalanced(u)  {
        u += String(w.removeFirst())
    }
    
    v = w
    
    if isCorrected(u) {
        return  u + solution(v)
    }
    
    return "(" + solution(v) + ")" + reversed(u)
}