[프로그래머스] Swift 튜플 2019 카카오 implementation

문제 :

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

코딩테스트 연습 - 튜플

 

분석 :

string 배열을 돌며 "{" 가 나타났을때 "}"가 나타나기 전까지 원소를 하나씩 담는다.

result array에 값이 없을때만 원소를 담아 리턴하도록 했는데 이 부분에서 O(n^2)을 사용해 되게 비효율적이라는 생각이 들었다. 

다른 사람들 풀이를 보니 component 또는 split를 활용해 쉽게 배열을 분리하고 계산하는것 같았다.

 

Swift 코드 :

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

import Foundation

func solution(_ s:String) -> [Int] {
    var str = s
    var reArr = Array(repeating: [Int](), count: 501)
    str.removeFirst()
    str.removeLast()
    var arr = Array(str)
    var result = [Int]()
    var i = 0
    var length = 0
    
    while i < arr.count {
        if arr[i] == "{" {
            length += 1
            var number = ""
            while arr[i] != "}"{
                i += 1
                if arr[i] == "}" {
                    reArr[length].append(Int(number)!)
                }
                else if arr[i] == "," {
                    reArr[length].append(Int(number)!)
                    number = ""
                }

                if 48 <= arr[i].asciiValue! && arr[i].asciiValue! < 57 {
                    number += "\(arr[i])"
                }
            }
        }
        i += 1
    }
    reArr = reArr.filter { !$0.isEmpty }
    reArr.sort { $0.count < $1.count }
    
    for i in reArr {
        if i.isEmpty { continue }
        if i.count == 1 { result.append(i[0]) }
        else {
            for num in i {
                if !result.contains(num) {
                    result.append(num)
                }
            }
        }
    }
    return result
}

 

다른 사람 풀이 :

import Foundation
func solution(_ s:String) -> [Int] {
    var s = s
    var answer = [Int]()
    s.removeFirst(2)
    s.removeLast(2)

    s.components(separatedBy: "},{")
        .map { $0.components(separatedBy: ",").map { Int($0)! } }
        .sorted { $0.count < $1.count }
        .forEach {
            $0.forEach {
                if !answer.contains($0) {
                    answer.append($0)
                }
            }
    }
    return answer
}

 

결과 :