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

프로그래머스

---

성공 풀이

import Foundation

func solution(_ p:String) -> String {
    
    var result = ""
    
    func correctCheck(_ s: String) -> Bool {
        var cStack = [String]()
        let strArr = s.map {String($0)}
        for i in 0..<s.count {
            if strArr[i]=="(" {
                cStack.append(strArr[i])
            } else {
                if cStack.isEmpty || cStack.last != "(" {
                    return false
                } else {
                    cStack.removeLast()
                }
            }
        }
        return cStack.isEmpty
    }
    
    func balanceCheck(_ w: String) -> String {
        if w=="" {return ""}
        var stack = [String]()
        var count = (0,0)
        var wArr = w.map {String($0)}
        var u = "", v = ""
        
        for i in 0..<wArr.count { // u,v 구하기
            if wArr[i] == "(" {
                count.0 += 1
            } else {
                count.1 += 1
            }
            stack.append(wArr[i])
            if count.0 == count.1 {
                u = stack.joined()
                v = wArr.suffix(wArr.count-i-1).joined()
                print("u: \(u), v: \(v)")
                break
             }
        }
        
        if u.first == "(" { // 원래는 만들었던 correctCheck 함수를 썻지만, 풀이상 u의 첫번째가 "("인지만 확인해도 됨.
            result += u
            return u+balanceCheck(v)
        } else {
            var temp = u.map {String($0)}
            temp.removeFirst()
            temp.removeLast()
            for i in 0..<temp.count {
                if temp[i] == "(" {
                    temp[i] = ")"
                } else {
                    temp[i] = "("
                }
            }
            return "(" + balanceCheck(v) + ")" + temp.joined()
        }
    }
    print(result)
    return balanceCheck(p)
}

풀이 키워드

- 지문의 예시를 잘 읽어보고 그대로 잘 구현해내면 되는 문제
- 하지만 나는 2시간이나 걸린 문제^^

---