[Swift - 프로그래머스] 불량 사용자

---

🤔 풀이 과정

user_id와 banned_id가 String 배열의 형태로 주어지게됩니다. banned_id를 기준으로 user_id를 탐색하며 매칭되는 id들의 인덱스들을 묶어서 저장해준뒤 매칭 가능한 모든 케이스의 수를 구하는 문제였습니다. 조합은 재귀함수를 통해서 구해보았으며 하나의 banned_id와 매칭되는 user_id의 인덱스들을 오름차순으로 정렬한 뒤 Set에 저장하여 중복을 제거하였습니다. 

 

---

📦  문제 풀이 & 코드

import Foundation

func solution(_ user_id: [String], _ banned_id: [String]) -> Int {

    // 인덱스끼리 매칭, 배열의 인덱스는 banned_id의 인덱스,값은 매칭되는 user_id들의 인덱스 배열
    var matchUpWithIndex = [[Int]](repeating: [Int](), count: banned_id.count)
    
    // Set은 가능한 모든 케이스들을 저장
    var matchUpCases = Set<[Int]>()
    
    // "*" 문자를 체크하며 매칭해보는 함수
    func matching(_ user_id: String, _ banned_id: String) -> Bool {
        if user_id.count != banned_id.count {
            return false
        }
        
        let tempUser = user_id.map{$0}
        let tempBanned = banned_id.map{$0}
        
        for idx in 0..<tempUser.count {
            if tempUser[idx] != tempBanned[idx] && tempBanned[idx] != "*" {
                return false
            }
        }
        
        return true
    }
    
    // 재귀함수를 통해 조합들을 구하여 matchUpCases에 저장.
    func combination(_ curBannedIdx: Int, selected: [Int]) {
        if curBannedIdx == banned_id.count {
        
            // ex) 0번째 banned_id와 [3, 2, 1] 매칭 == [1, 2, 3] 매칭
            // 매칭되는 user 인덱스들의 순서는 정렬하여 같은것으로 처리
            
            matchUpCases.insert(selected.sorted(by: <))
            
        } else {
            var temp = selected
            
            for userIdx in matchUpWithIndex[curBannedIdx] {
                if temp.contains(userIdx) == false {
                    temp.append(userIdx)
                    combination(curBannedIdx + 1, selected: temp)
                    temp.removeLast()
                }
            }
        }
    }
    
    for bannedIndex in 0..<banned_id.count {
        for userIndex in 0..<user_id.count {
            if matching(user_id[userIndex], banned_id[bannedIndex]) {
                matchUpWithIndex[bannedIndex].append(userIndex)
            }
        }
    }
    
    combination(0, selected: [])
    
    return matchUpCases.count
}

• 문제 링크: https://programmers.co.kr/learn/courses/30/lessons/64064

코딩테스트 연습 - 불량 사용자