[백준] (Swift) 1541번 - 잃어버린 괄호 (그리디)

공중전화가 생각나는 문제번호다,,, 10초였나.. 쨌든 요금 안나가게 할라고 몇초만에 빨리말하고 전화 끊고 그랬었는데...

 

🟠 문제

https://www.acmicpc.net/problem/1541

1541번: 잃어버린 괄호

'

🟠 나의 풀이 

우선, 시간초가 2초라서, 여유롭게 풀었다. 2억번의 연산이 발생해도 좋다는 뜻이기 때문에 시간복잡도를 그닥? 생각하지 않고 풀었다.

문제는 생각보다 간단하다. 더하기와 빼기 밖에 없기 때문에, 마이너스해주는 수가 가장 커지면, 값은 최소로 출력할 수 있다. 즉, + 연산을 모두 수행해주고 (더해서 가장 크게 만들어주고) - 연산을 수행해주면 된다! 그럼결국 빼는 수가 점점 커지게 되서 최소의 수가 출력될 수 있다.

🟠 정답코드

굿,, 

import Foundation

private func solution(_ input: String) -> Int {
    let input = input.split(separator: "-").map{ $0 }
    var minusArr = [Int]()

    for i in 0..<input.count {
        if !input[i].contains("+") {
            minusArr.append(Int(input[i])!)
            continue
        }

        let temp = input[i].split(separator: "+").map{ Int($0)! }.reduce(0,+)
        minusArr.append(temp)
    }

    if minusArr.count > 1 {
        var total = minusArr[0]
        for j in 1..<minusArr.count {
            total -= minusArr[j]
        }
        return total
    } else {
        return minusArr[0]
    }
}

let input = readLine()!
print(solution(input))