[swift] 백준 - 2504번: 괄호의 값

2504번: 괄호의 값

---

성공 풀이

let a = readLine()!.map { String($0) }
let value = ["(": 2, "[": 3]
let pair = [")": "(", "]": "["]
var stack = [String]()
var temp = 1
var result = 0

for i in 0..<a.count {
    if a[i]=="(" || a[i]=="[" {
        temp *= value[a[i]]!
        stack.append(a[i])
    } else {
        if stack.isEmpty || stack.last! != pair[a[i]] {
            result = 0
            break
        } else {
            if a[i-1]==pair[a[i]] { result += temp }
            temp /= value[stack.removeLast()]!
        }
    }
}
print(stack.isEmpty ? result : 0)

풀이 키워드

예제 :  (()[[]])([])

괄호가 열릴 때마다 2 또는 3의 값을 곱해서 누적시킵니다.
예제를 가지고 예를 들어보겠습니다.

첫 번째 문자 '('가 들어오면 temp 변수에 2를 저장하고 stack배열에 괄호를 추가합니다.
처음 temp변수는 1로 초기화시켜줬습니다.

temp = 2​

다음 문자 '('도 열리는 괄호입니다. 기존 temp값과 2를 곱해서 temp에 저장하고 괄호를 추가합니다.
해당 계산은 괄호 안에 괄호가 있을 때 곱해주는 성질 때문에 괄호가 열릴 때마다 temp값에 곱하게 되는 겁니다.

temp = 4

다음은 닫히는 괄호가 나옵니다.
닫힌 괄호가 나올 때 stack이 비어있거나 stack마지막 괄호가 현재 닫힌 괄호의 짝이 아니라면
올바르지 못한 괄호열이기 때문에 result에 0을 넣고 break 시킵니다.

stack의 마지막 괄호가 현재 닫힌 문자와 짝 일 때
a배열에서 앞의 문자가 현재 문자와 짝을 이룬다면 (현재 ')'닫힌 배열의 앞의 문자는'('로 짝을 이루고 있음)
temp값을 result값에 더하게 됩니다.

temp = 4
result = 4

a배열의 앞의 문자가 현재 문자와 짝을 이룰 때만 온전히 해당 괄호가 가지고 있는 값이 계산됩니다.
예) (()[[]])([]) , 주황색이 칠해진 부분만 현재 temp값을 result에 더해주게 됨.

다음 닫힌 배열의 value만큼 temp값을 나눠주고, 스택에서 지워주면 됩니다.

temp = 2
reesult = 4

위 공식을 반복문이 끝날 때까지 적용한다면

모양:(, temp:2, result:0
모양:(, temp:4, result:0
모양:), temp:2, result:4
모양:[, temp:6, result:4
모양:[, temp:18, result:4
모양:], temp:6, result:22
모양:], temp:2, result:22
모양:), temp:1, result:22
모양:(, temp:2, result:22
모양:[, temp:6, result:22
모양:], temp:2, result:28
모양:), temp:1, result:28

위 순서대로 temp값과 result값이 바뀌게 됩니다.

---