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

프로그래머스 LV2. 수식 최대화

 

단순 구현이고 정확히만 하면 되는데, 종이 없이 하려고 연습중인데 조금 오래걸리는거 같다.

모듈화 등등 ... 이게 개발 습관이라서 알고리즘 풀때는 요런 부분보다는 로직에 집중.

 

 

 

샘플 코드

알고리즘 접근법에 대해서 처음에는 계산기 알고리즘을 사용하려고 했다.

중위 후위 등 구글에 치면 다양한 알고리즘이 나옴.

 

하지만 난, 꼭 그런걸 정량적으로 알고 있어야 문제를 푸는건 불만이었기에 다른 로직으로 내 방식대로 풀어보았다.

priority를 6번 반복하는 이유는 어차피 연산자 3개로 우선순위를 매기는 경우의 수가 6가지 밖에 되지 않는다.

 

두번째는 while문인데, 연산자를 찾아서 그 연산자의 앞뒤로 해당하는 연산을 처리하고

결과값을 배열에 넣어주고 이미 계산한 숫자와 연산자를 배열에서 모두 삭제해줌으로써 새로운 배열을 세팅한다.

그렇게 우선순위에 따라서 처리하고나서 답을 구하면 끝.

 

마지막으로 출력부에서 알아두면 좋은 부분인데, max안에 abs를 같이 넣었더니 시간 초과가 나서 이는 밖으로 빼서 계산했다.

 

 

//
//  67257.swift
//  Algorithm
//
//  Created by Hamlit Jason on 2022/04/16.
//
// https://programmers.co.kr/learn/courses/30/lessons/67257
import Foundation

struct p67257 {
    static func run() {
        print(p67257.solution("100-200*300-500+20")) // 60420
        // * > + > - 순서 : 100 - 60000 - 520  = 60420
        // array : 100, -, 60000, -, 520
//                print(p67257.solution("50*6-3*2")) // 300
        // - > * 순서: 50, *, 3, * 2
        
        
    }
    
    static func solution(_ expression:String) -> Int64 {
        
        let numbers = expression.split { $0 == "*" || $0 == "+" || $0 == "-" }.map { String($0) }
        let operators = expression.split { $0.isNumber }.map { String($0) }
        
        
        var list = [String]()
        for i in 0..<operators.count {
            list.append(numbers[i])
            list.append(operators[i])
        }
        list.append(numbers.last!)
//        print("list \(list)")
        
        // 오퍼레이터의 우선순위를 설정합니다.
        let priority = [
            ("*","-","+"),
            ("*","+","-"),
            ("+","*","-"),
            ("+","-","*"),
            ("-","+","*"),
            ("-","*","+")
        ]
        
        var sumArray = [Int](repeating: 0, count: 6)
        var copyList = list
        
        for i in 0..<priority.count {
            copyList = list
            let first = priority[i].0
            let second = priority[i].1
            let third = priority[i].2
            
            while copyList.contains(first) {
                let index = copyList.firstIndex(of: first)!
                
                sumArray[i] = caculate(a: Int(copyList[index-1])!, b: Int(copyList[index+1])!, oper: first)
                
                copyList.insert(String(sumArray[i]), at: index+2)
                
                copyList.remove(at: index-1)
                copyList.remove(at: index-1)
                copyList.remove(at: index-1)
            }
            
            while copyList.contains(second) {
                let index = copyList.firstIndex(of: second)!
                sumArray[i] = caculate(a: Int(copyList[index-1])!, b: Int(copyList[index+1])!, oper: second)
                copyList.insert(String(sumArray[i]), at: index+2)
                
                copyList.remove(at: index-1)
                copyList.remove(at: index-1)
                copyList.remove(at: index-1)
                
            }
            
            while copyList.contains(third) {
                let index = copyList.firstIndex(of: third)!
                sumArray[i] = caculate(a: Int(copyList[index-1])!, b: Int(copyList[index+1])!, oper: third)
                copyList.insert(String(sumArray[i]), at: index+2)
               
                copyList.remove(at: index-1)
                copyList.remove(at: index-1)
                copyList.remove(at: index-1)
            }
            
        }
//        print(sumArray)
        var maxValue = 0
        sumArray.forEach { value in
            let value = abs(value)
            maxValue = max(value, maxValue)
        }
        
        return Int64(maxValue)
    }
    
    static func caculate(a: Int, b: Int, oper: String) -> Int {
        switch oper {
        case "*": return a * b
        case "-": return a - b
        case "+": return a + b
        default: return -1
        }
    }
}

 

 

 

성능이 매우 좋다!