[Algorithm] 혼자 놀기의 달인

[hwangJi-dev]

💬 문제

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

💬 Idea

• cardDict에 순서 - 카드 숫자를 저장해놓는다.
• 이후 첫번째 key부터 그룹지을 수 있는 만큼 (해당 숫자가 열려있을 때까지) 반복하면서 각 그룹을 찾아 ans에 그 개수를 저장한다.
• ans를 내림차순 정렬한 뒤
  • 그룹 개수가 2개 이상이라면 최댓값, 그다음 최댓값을 곱한 숫자를 return한다
  • 그룹 개수가 2개 미만이라면 0을 return한다.

💬 풀이

import Foundation

func solution(_ cards:[Int]) -> Int {
    var cardDict: [Int: Int] = [:]
    
    for (idx, i) in cards.enumerated() {
        cardDict[idx + 1] = i
    }
    
    var ans: [Int] = []
    
    func cycle(_ start: Int) -> Int {
        var queue: [Int] = []
        var start = start
        
        while cardDict[start] != nil {
            let next = cardDict[start]!
            queue.append(start)
            cardDict.removeValue(forKey: start)
            start = next
        }
        
        return queue.count
    }
    
    while !cardDict.isEmpty {
        ans.append(cycle(cardDict.keys.first!))
    }

    ans = ans.sorted(by: >)
    
    return ans.count >= 2 ? ans[0] * ans[1] : 0
}