[Swift] 2019 카카오 개발자 겨울 인턴쉽 불량 사용자

Problem

 

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

---

Solution

 

1. 불량 사용자와 유저 아이디를 비교한다.

 

func isEqual(userId:String,bannedId:String) -> Bool {
    if userId.count != bannedId.count { return false }
    
    let uid = userId.map{String($0)}
    let bid = bannedId.map{String($0)}
    
    for (i,b) in bid.enumerated() {
        if b != "*" && uid[i] != bid[i]  {
            return false
        }
    }
    return true
}

 

2. 비교한 아이디 중 해당하는 아이디를 따로 모아놓는다.

 

func findBanIdEqualUserId(userId:[String],banId:[String]) -> [[Int]] {
    var banArray = Array(repeating: [Int](), count: banId.count)
    
    for (i,uid) in userId.enumerated() {
        for (j,bid) in banId.enumerated() {
            if isEqual(userId: uid, bannedId: bid) {
                banArray[j].append(i)
            }
        }
    }
    return banArray
}

 

3. 불량사용자들의 모든 경우의 수를 구한다.

 

func insertBanIdsToResult(banArray:[[Int]],banIds:Set<Int>,result:inout Set<[Int]>,current:Int) {
    for id in banArray[current] {
        var newBanIds = banIds
        newBanIds.insert(id)
        if current != banArray.count - 1 {
            insertBanIdsToResult(banArray:banArray,banIds: newBanIds,result:&result ,current: current+1)
        }else if newBanIds.count == banArray.count {
            result.insert(newBanIds.sorted())
        }
    }
}

 

4. 경우의 수의 갯수를 반환한다.

 

return result.count

---

Source Code