[프로그래머스 LV.0] 문자열 계산하기

프로그래머스 LV.0 모음

문자열 계산하기

문제 설명

my_string은 "3 + 5"처럼 문자열로 된 수식입니다. 문자열 my_string이 매개변수로 주어질 때, 수식을 계산한 값을 return 하는 solution 함수를 완성해 주세요.

제한 사항

• 연산자는 +, -만 존재합니다.
• 문자열의 시작과 끝에는 공백이 없습니다.
• 0으로 시작하는 숫자는 주어지지 않습니다.
• 잘못된 수식은 주어지지 않습니다.
• 5 ≤ my_string의 길이 ≤ 100
• my_string을 계산한 결괏값은 1 이상 100,000 이하입니다.
  • my_string의 중간 계산 값은 -100,000 이상 100,000 이하입니다.
  • 계산에 사용하는 숫자는 1 이상 20,000 이하인 자연수입니다.
  • my_string에는 연산자가 적어도 하나 포함되어 있습니다.
• return type 은 정수형입니다.
• my_string의 숫자와 연산자는 공백 하나로 구분되어 있습니다.

입출력 예

my_string	result
"3 + 4"	7

입출력 예 설명

입출력 예 #1

• 3 + 4 = 7을 return 합니다.

---

제출

import Foundation

func solution(_ my_string:String) -> Int {
    var string_value = my_string.split(separator: " ").map{String($0)}
    var answer = Int(string_value[0])!
    
    for index in stride(from: 1, to: string_value.count, by: 2) {
        if string_value[index] == "+"{
            answer += Int(string_value[index+1])!
        }else{
            answer -= Int(string_value[index+1])!
        }
    }
    return answer
}

입력받은 문자열을 공백 기준으로 나눠 String 배열로 저장한다.
answer에 배열의 첫 번째 값을 Int로 변환해 넣어준다.

for문을 사용해 index를 1부터 배열의 크기까지 2씩 증가시키며 반복한다. (숫자, 연산자, 숫자 순서로 반복되기 때문)
연산자가 + 라면 다음 인덱스의 숫자와 덧셈을, 아니라면 뺄셈을 한다.

다른 풀이

import Foundation

func solution(_ my_string: String) -> Int {
    return my_string.replacingOccurrences(of: "- ", with: "-")
        .replacingOccurrences(of: "+", with: "")
        .split(separator: " ")
        .map { Int($0)! }
        .reduce(0, +)
}

replacingOccurrences()을 이용해 "- "는 "-"로, "+"는 지워준다. (ex. 5 + 2 - 3 => 5 2 -3)
이후 공백을 기준으로 나눈 후 map 함수를 사용해 Int형 배열로 만든다.
reduce 함수를 사용해 배열을 더해준다.