[스위프트 알고리즘] 프로그래머스: 괄호 회전하기

문제

다음 규칙을 지키는 문자열을 올바른 괄호 문자열이라고 정의합니다.

• (), [], {} 는 모두 올바른 괄호 문자열입니다.
• 만약 A가 올바른 괄호 문자열이라면, (A), [A], {A} 도 올바른 괄호 문자열입니다. 예를 들어, [] 가 올바른 괄호 문자열이므로, ([]) 도 올바른 괄호 문자열입니다.
• 만약 A, B가 올바른 괄호 문자열이라면, AB 도 올바른 괄호 문자열입니다. 예를 들어, {} 와 ([]) 가 올바른 괄호 문자열이므로, {}([]) 도 올바른 괄호 문자열입니다.

대괄호, 중괄호, 그리고 소괄호로 이루어진 문자열 s가 매개변수로 주어집니다. 이 s를 왼쪽으로 x (0 ≤ x < (s의 길이)) 칸만큼 회전시켰을 때 s가 올바른 괄호 문자열이 되게 하는 x의 개수를 return 하도록 solution 함수를 완성해주세요.

풀이

스택을 사용해서 풀이했습니다. 조건문이 너무 길어지는 것을 최적화하고 싶어서 딕셔너리를 사용해서 스택 가장 상단에 있는 문자가 이번에 넣은 문자의 짝인지 확인할 수 있도록 했습니다.

코드

import Foundation

func isBalanced(_ s: [Character]) -> Bool {
    var stack: [Character] = []
    let braces: [Character:Character] = ["{":"}", "[":"]", "(":")"]
    for i in s {
        if !stack.isEmpty && braces[stack.last!] == i {
            stack.removeLast()
        } else {
            stack.append(i)
        }
    }
    return stack.isEmpty
}

func solution(_ s:String) -> Int {
    var str = Array(s)
    print(type(of: str))
    var answer = 0
    for _ in str {
        str.append(str.removeFirst())
        if isBalanced(str) {
            answer += 1
        }
    }

    return answer
}