[백준 5613] 계산기 프로그램

5613번: 계산기 프로그램

첫 입력은 그냥 저장되고,

연산자는 입력마다 저장하고,

그 다음부터 숫자가 들어올 때 마다 결과값이 바뀌어야 한다.

예를 들어 1 + 1 을 계산할 때

• 1은 첫 입력이므로 결과값에 그냥 저장
• • 은 연산자이므로 연산자로 저장
• 1이 들어오면 기존 결과값에 연산자를 활용해 1을 계산해야 함 → 결과값 갱신

풀이

var currentResult = 0
var currentOperator = ""
while let input = readLine() {
    guard let intInput = Int(input) else { // 입력이 연산자인지 숫자인지 판단
        if input == "=" {
            break
        } else {
            currentOperator = input
            continue
        }
    }
    switch currentOperator {
        case "+":
        currentResult += intInput
        case "-":
        currentResult -= intInput
        case "*":
        currentResult *= intInput
        case "/":
        currentResult /= intInput
        default: // ""인 경우 = 첫 입력
        currentResult = intInput
        continue
    }
}
print(currentResult)

다른 풀이

나는 입력 한 줄을 반복단위로 삼았는데, 이렇게 하신 분도 있었다!

• 첫 입력은 그냥 받기 (readLine())
• 이후 입력부터 “연산자, 숫자” 단위로 반복해 계산

시간복잡도는 같지만, 이쪽이 메모리를 조금 덜 사용한다.

var num = Int(readLine()!)!
var sum = num
while true {
    let operators = readLine()!
    if operators == "=" {
        break
    }
    num = Int(readLine()!)!
    switch operators {
    case "+":
        sum += num
    case "-":
        sum -= num
    case "*":
        sum *= num
    case "/":
        sum /= num
    default:
        break
    }
}
print(sum)