[프로그래머스] (Swift) Kakao - 수식 최대화 (Lv.2)

🟠 문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/67257

프로그래머스

 

🟠 나의 풀이

우선순위를 부여하고, 그에 따른 연산을 하고,, 복잡해 보이지만 이해만 하면 단순한 문제였다.

우선 나는 주어진 expression을 '숫자'와 '연산자'로 이루어진 배열을 만들어야겠다고 생각했다. 이 부분에서 어떻게 바꾸면 좋을지 생각하다가 좋은 코드를 발견해서 조금 참고했다.

let num = "1234"
if num.isNumber {
	print(true)
}

isNumber 이라는 메서드를 이용하면, 해당 문자가 숫자인지 판단하여 Bool값으로 리턴한다. 이 것을 이용해서 expression에 있는 '한 문자'씩꺼내어 number면, num에 차곡차곡 쌓아주고, 아니라면 연산자이므로 지금까지 쌓인 num을 배열에 넣어주고 연산자까지 넣어주는 형식으로 진행했다. 

 

그리고 이제 어떻게 이 배열을 연산할 것인가 생각해야한다.

["100", "-", "200", "*", "300", "-", "500", "+", "20"]

이런 배열에서 '연산자'를 기준으로 앞뒤 숫자를 계산한다. 아래와 같이 생각할 수 있다.

 

그래서 각 operation 마다 계산된 result를 remove 세번을 통해 기존값으 삭제하고, 한번 append 해주어서 newArr를 갱신해 나갔다.

더 자세하거나 상세한 것은 코드 내부 주석으로 달아두었다.

 

🟠 정답코드

https://github.com/deslog/Algorithm/blob/main/Algorithm/Programmers/kakao_%EC%88%98%EC%8B%9D%20%EC%B5%9C%EB%8C%80%ED%99%94/main.swift

GitHub - deslog/Algorithm

//
//  main.swift
//  Algorithm
//
//  Created by LeeJiSoo on 2022/09/01.
//

import Foundation

let expression = "100-200*300-500+20"
// result = 60420

func solution(_ expression:String) -> Int {

    // 최종 리턴할 result
    var resultList = [Int]()

    // input을 List로 바꿔주기
    var num = ""
    var arr = [String]()
    expression.forEach { char in
        if char.isNumber {
            num += String(char)
        } else {
            arr.append(num) // 지금까지 쌓아놨던 숫자 추가
            arr.append(String(char)) // 연산자 추가
            num = "" //num 초기화
        }
    }
    arr.append(num) // 마지막 숫자 넣어줘야해 (이거때문에 시간 개잡아먹었네)
    print(arr)

    // 연산자
    let operatorList = [["+", "-", "*"], ["+", "*", "-"],
                        ["-", "+", "*"], ["-", "*", "+"],
                        ["*", "-", "+"], ["*", "+", "-"]]

    var newArr = arr

    operatorList.forEach { operators in
        // newArr에 1개만 남을때까지 -> 연산하면 결국 하나의 결과만 남으니까
        while newArr.count > 1 {
            operators.forEach { op in
                while newArr.contains(op) { // 연산자 있는대로 계산해야하니까 while문
                    let i =  newArr.firstIndex(of: String(op))! // op 인덱스 번호 찾아
                    var result = 0

                    if op == "+" {
                        result = Int(newArr[i-1])! + Int(newArr[i+1])!
                    } else if op == "-" {
                        result = Int(newArr[i-1])! - Int(newArr[i+1])!
                    } else if op == "*" {
                        result = Int(newArr[i-1])! * Int(newArr[i+1])!
                    }
                    // 연산자와 앞뒤 숫자 지워주고, 그자리에 넣어야해
                    newArr.remove(at: i-1)
                    newArr.remove(at: i-1)
                    newArr.remove(at: i-1)
                    newArr.insert(String(result), at: i-1)
                }
            }
        } // while newArr.count > 1 종료지점

        let temp = Int(newArr[0])!
        resultList.append(abs(temp))
        newArr = arr // 다음 연산자 세트를 위해서 다시 초기화
    }

    return resultList.max()!
}

print(solution(expression))