본문 바로가기

# 02/Swift - CTP

[Swift] 코딩테스트 연습! Lv2. 괄호 회전하기

반응형
/* 괄호 회전하기

 - 다음 규칙을 지키는 문자열을 올바른 괄호 문자열이라고 정의합니다.
 
 (), [], {} 는 모두 올바른 괄호 문자열입니다.
 만약 A가 올바른 괄호 문자열이라면, (A), [A], {A} 도 올바른 괄호 문자열입니다. 예를 들어, [] 가 올바른 괄호 문자열이므로, ([]) 도 올바른 괄호 문자열입니다.
 만약 A, B가 올바른 괄호 문자열이라면, AB 도 올바른 괄호 문자열입니다. 예를 들어, {} 와 ([]) 가 올바른 괄호 문자열이므로, {}([]) 도 올바른 괄호 문자열입니다.
 대괄호, 중괄호, 그리고 소괄호로 이루어진 문자열 s가 매개변수로 주어집니다. 이 s를 왼쪽으로 x (0 ≤ x < (s의 길이)) 칸만큼 회전시켰을 때 s가 올바른 괄호 문자열이 되게 하는 x의 개수를 return 하도록 solution 함수를 완성해주세요.

 제한사항
 s의 길이는 1 이상 1,000 이하입니다.
*/
func solution19(_ s:String) -> Int {
    if s.count == 1 {
        return 0
    }
    var result:Int = 0
    var charList = Array(s)
    for _ in 0...s.count-1 {
        charList.append(charList.first!)
        charList.removeFirst()
        
        var list:[Character] = []
        var isMatch = true
        
        for char in charList {
            if char == "[" || char == "{" || char == "("{
                list.append(char)
            } else if char == "]" {
                if list.contains("[") {
                    let index = list.lastIndex(of: "[")!
                    if (list.lastIndex(of: "{") ?? 0) > index
                        || (list.lastIndex(of: "(") ?? 0) > index {
                        isMatch = false
                        break
                    }
                    list.remove(at: index)
                } else {
                    isMatch = false
                    break
                }
            } else if char == "}" {
                if list.contains("{") {
                    let index = list.lastIndex(of: "{")!
                    if (list.lastIndex(of: "[") ?? 0) > index
                        || (list.lastIndex(of: "(") ?? 0) > index {
                        isMatch = false
                        break
                    }
                    list.remove(at: index)
                } else {
                    isMatch = false
                    break
                }
            } else if char == ")" {
                if list.contains("(") {
                    let index = list.lastIndex(of: "(")!
                    if (list.lastIndex(of: "[") ?? 0) > index
                        || (list.lastIndex(of: "{") ?? 0) > index {
                        isMatch = false
                        break
                    }
                    list.remove(at: index)
                } else {
                    isMatch = false
                    break
                }
            }
        }
        if list.count == 0 && isMatch {
            result += 1
        }
    }
    return result
}
반응형