[Swift] 백준 1541 - 잃어버린 괄호

문제 바로가기

풀이

실버2지만,, 어려웠다 🥲
여기서 포인트는

• 괄호의 수에는 제한이 없다
• 값을 최소로 만들려고 한다.

이 두가지 이다.

따라서 연산자가 "-"일 때마다 괄호를 넣어 주면 된다.
이렇게 하기 위해서 입력받은 값을 "-"를 기준으로 나눠주면 된다.

예를 들어 "55-50+40-60+20"을 입력받아
"-"를 기준으로 나누면
["55","50+40","60+20"] 가 된다.

원소 들을 각각 모두 더해주고
최종 결과 값은 55 - [(50+40)+(60+20)] 이 된다.

제출 코드

import Foundation

let minusSplit = readLine()!.split(separator:"-").map{String($0)}
let plus = minusSplit[0].split(separator:"+").map{Int(String($0))!}

var result = 0
for i in plus {
	result += i
}
var minusTotal = 0
for i in 1..<minusSplit.count {
	let minus = minusSplit[i].split(separator:"+").map{Int(String($0))!}
    
    for j in minus {
    	minusTotal += j
    }
}

print(result - minusTotal)

• minusSplit[0]는 + 연산자만 있으므로 바로 result에 더해준다.
• minusSplit[1..나머지]에 만약 "+"가 없더라도 Int(-60) = -60이니 신경쓰지 않아도 된다.