[프로그래머스] Swift 수식 최대화

문제 :

https://programmers.co.kr/learn/courses/30/lessons/67257?language=swift 

코딩테스트 연습 - 수식 최대화

난이도 : level2

 

분석 :

*, -, + 3개의 연산자를 가지고 수식을 최대화 하는 연산자의 우선순위를 적용해 최대값을 리턴하는 문제다.

나는 수식이 3개밖에 없어서 (3! => 6) 가능한 경우의 수를 손코딩하였다.

그리고 연산자가 등장하는 순서대로 op배열에 담았고 

가능한 우선순위를 모두 계산해 result배열에서 계산한 값은 삭제하고 계산하여 원소가 한개 남았을 경우 최대값을 업데이트 해주었다.

 

Swift 코드 :

//
//  main.swift
//  SOMA👩🏻‍💻
//
//  Created by JoSoJeong on 2022/04/14.
//

import Foundation
func calc(_ n1: Int64, _ n2: Int64, _ op: String) -> Int64 {
    switch op {
    case "+":
        return n1 + n2
    case "*":
        return n1 * n2
    case "-":
        return n1 - n2
    default:
        return 0
    }
}

func solution(_ expression:String) -> Int64 {
    let priority: [[String]] = [["*", "+", "-"], ["*", "-", "+"], ["+", "*", "-"], ["+", "-", "*"], ["-", "+", "*"], ["-", "*", "+"]]
    var str = expression
    var arr = str.components(separatedBy: ["-", "*", "+"]).map { Int64(String($0))! }
    var op: [String] = []
    
    //op 순서대로 계산
    for i in str {
        if i == "-" {
            op.append("-")
        }else if i == "*" {
            op.append("*")
        }else if i == "+" {
            op.append("+")
        }
    }
    
    var answer:Int64 = 0
    for i in priority {
        var now = i
        var result = arr
        var nowOp = op
        for ii in now {
            while nowOp.contains(ii) {
                if let index = nowOp.firstIndex(of: ii) {
                    result[index] = calc(result[index], result[index+1], ii)
                    result.remove(at: index+1) // 계산한 숫자 지우기
                    nowOp.remove(at: index)
                }
            }
            
        }
        if result.count != 1 { continue }
        if result.first! < 0 { //절대값
            answer = max(answer, -result.first!)
        }else {
            answer = max(answer, result.first!)
        }
        
    }
    return answer
}

 

실행 결과 :