불량 사용자 - Kotlin (DFS)

프로그래머스

문제 설명

개발팀 내에서 이벤트 개발을 담당하고 있는 "무지"는 최근 진행된 카카오이모티콘 이벤트에 비정상적인 방법으로 당첨을 시도한 응모자들을 발견하였습니다. 이런 응모자들을 따로 모아 불량 사용자라는 이름으로 목록을 만들어서 당첨 처리 시 제외하도록 이벤트 당첨자 담당자인 "프로도" 에게 전달하려고 합니다. 이 때 개인정보 보호을 위해 사용자 아이디 중 일부 문자를 '*' 문자로 가려서 전달했습니다. 가리고자 하는 문자 하나에 '*' 문자 하나를 사용하였고 아이디 당 최소 하나 이상의 '*' 문자를 사용하였습니다.
"무지"와 "프로도"는 불량 사용자 목록에 매핑된 응모자 아이디를 제재 아이디 라고 부르기로 하였습니다.

예를 들어, 이벤트에 응모한 전체 사용자 아이디 목록이 다음과 같다면

 

다음과 같이 불량 사용자 아이디 목록이 전달된 경우,

불량 사용자에 매핑되어 당첨에서 제외되어야 야 할 제재 아이디 목록은 다음과 같이 두 가지 경우가 있을 수 있습니다.

 

이벤트 응모자 아이디 목록이 담긴 배열 user_id와 불량 사용자 아이디 목록이 담긴 배열 banned_id가 매개변수로 주어질 때, 당첨에서 제외되어야 할 제재 아이디 목록은 몇가지 경우의 수가 가능한 지 return 하도록 solution 함수를 완성해주세요.

 

 

나의 풀이

이 문제는 DFS를 이용해서 탐색하는 것과 제재할 수 있는 아이디인지 체크하는 부분 2가지로 나눌 수 있다.

 

DFS를 사용해서 탐색하는 부분은 생각보다 간단하다. 불량 사용자 모음의 길이만큼 탐색하고 중복을 허용하지 않기 위해 최대 길이만큼 탐색을 완료했을 시 set에 String으로 변환하여 저장한다.

 

아이디를 비교하는 부분은 간단하다. 글자가 *가 아닐 때, 만약 두 글자가 일치하지 않는다면 불량 아이디 조건을 만족하지 못하는 것임으로 바로 false를 반환하고 모든 탐색을 완료하면 true를 반환하면 된다.

 

class Solution {    
    val visitedSet = mutableSetOf<String>()
  
    fun solution(user_id: Array<String>, banned_id: Array<String>): Int {
        dfs(0, banned_id.size, IntArray(user_id.size), user_id, banned_id)     
        return visitedSet.size
    }
    
    fun dfs(depth: Int, target: Int, visited: IntArray, user_id: Array<String>, banned_id: Array<String>) {
        if (depth == target) {
            visitedSet.add(visited.contentToString())
            return
        }
        
        for (i in user_id.indices) {
            if (visited[i] == 0 && check(user_id[i], banned_id[depth])) {
                visited[i] = 1
                dfs(depth + 1, target, visited, user_id, banned_id)
                visited[i] = 0
            }
        }
    }
    
    fun check(user: String, banned: String): Boolean {
        if (user.length != banned.length)
            return false
        
        for (i in 0 until user.length) {
            if (banned[i] != '*' && user[i] != banned[i])
                return false
        }
        return true
    }
}