[Python/Programmers] 불량 사용자

https://school.programmers.co.kr/learn/courses/30/lessons/64064

프로그래머스

문제 설명

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

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

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

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

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

풀이

Code

def solution(user_id, banned_id):
    # 1. 백트래킹 함수 정의
    def backtracking(idx, combination) :
        # 1-1. 종료 조건 설정
        if idx == n :
            combination.sort()
            if combination not in answer :
                answer.append(combination)
            return
        # 1-2. 현재 불량 사용자 아이디에 해당하는 아이디가 없을 경우
        if not possible_id[idx] :
            backtracking(idx+1, combination)
        # 1-3. 현재 불량 사용자 아이디에 해당하는 아이디가 있을 경우
        else :
            # 1-3-1. 
            for id in possible_id[idx] :
                # 다음 아이디가 현재 아이디 목록에 없을 경우 백트래킹 실행
                if id not in combination :
                    backtracking(idx+1, combination + [id])
    answer = []
    n = len(banned_id)
    # 2. 불량 사용자로 가능한 아이디 리스트 생성
    possible_id = []
    # 3.
    for b_id in banned_id :
        # 3-1. 현재 불량 아이디로 가능한 아이디 리스트 생성
        array = []
        # 3-2.
        for u_id in user_id :
            # 가능할 경우 append
            if len(u_id) == len(b_id) :
                for i in range(len(b_id)) :
                    if b_id[i] == '*' : continue
                    if u_id[i] != b_id[i] : break
                else :
                    array.append(u_id)
        # 3-3. 불량 사용자로 가능한 아이디 리스트에 append
        possible_id.append(array)
    # 4. 백트래킹 실행
    backtracking(0, [])
    # 5. 결과 리턴
    return len(answer)