[Python][프로그래머스] 64064_불량 사용자

2019 카카오 개발자 겨울 인턴십

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

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

문제 설명

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

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

 

응모자 아이디

frodo

fradi

crodo

abc123

frodoc

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

불량 사용자

fr*d*

abc1**

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

제재 아이디

frodo

abc123

제재 아이디

fradi

abc123

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

[제한사항]

• user_id 배열의 크기는 1 이상 8 이하입니다.
• user_id 배열 각 원소들의 값은 길이가 1 이상 8 이하인 문자열입니다.
  • 응모한 사용자 아이디들은 서로 중복되지 않습니다.
  • 응모한 사용자 아이디는 알파벳 소문자와 숫자로만으로 구성되어 있습니다.
• banned_id 배열의 크기는 1 이상 user_id 배열의 크기 이하입니다.
• banned_id 배열 각 원소들의 값은 길이가 1 이상 8 이하인 문자열입니다.
  • 불량 사용자 아이디는 알파벳 소문자와 숫자, 가리기 위한 문자 '*' 로만 이루어져 있습니다.
  • 불량 사용자 아이디는 '*' 문자를 하나 이상 포함하고 있습니다.
  • 불량 사용자 아이디 하나는 응모자 아이디 중 하나에 해당하고 같은 응모자 아이디가 중복해서 제재 아이디 목록에 들어가는 경우는 없습니다.
• 제재 아이디 목록들을 구했을 때 아이디들이 나열된 순서와 관계없이 아이디 목록의 내용이 동일하다면 같은 것으로 처리하여 하나로 세면 됩니다.

---

banned_id와 같은 user_id을 찾아내 조합(Combination)을 구하는 문제이다.

단, 문제에서 주의할 점이 banned_id에서 일부가 '*'로 정확한 값을 모르기 때문에 일반적으로 조합을 구하는 방식으로 풀이하면

user_id = ['aab', 'aaa', 'bbb', 'ccc'] / banned_id = ['aa*', 'aa*'] 와 같은 식으로 주어졌을 때 ['aab', 'aaa']와 ['aaa', 'aab']서로 다르다고 생각하며 2번 선택하게 된다는 것이다.

 

--> 해결으로 banned_id 만큼 user_id에서 조합을 선택한 뒤에 선택값을 기억하고 있으면 된다.

--> 이는 vi배열 대신에 비트연산자를 사용하고, 이 bit를 set에 저장하여 조합선택 후 해당 bit가 이미 set에 있는지 확인하면 된다.

 

def solution(user_id, banned_id): # 전체 아이디 N개 중에서 차단 아이디 M개의 조합 구하기
    
    global answer, ansSet
    
    answer= 0
    ansSet = set() # M개를 선택했을 때 선택여부 체크 bit의 상태를 추가
    comb(0, 0, user_id, banned_id) # 선택 개수, bit

    return answer
    
def comb(sel, bit, user_id, banned_id):
    
    global answer, ansSet
    
    if sel == len(banned_id): # M개 만큼 선택을 했을 때
        if  bit not in ansSet: # 선택여부 체크 bit가 이미 set에 존재하지 않는 경우에만 답으로 추가
            ansSet.add(bit)
            answer+=1
        return
    
    # user_id를 순회하며 sel번째 banned_id와 같은 값이 있을 때 이를 선택하고,
    # 중복선택이 되지 않도록 bit를 체크
    for i in range(len(user_id)):
        if bit & 1 << i == 0 and chk(user_id[i], banned_id[sel]):
            comb(sel+1, bit | 1 << i, user_id, banned_id)

def chk(user, banned): # user아이디와 banned아이디가 같을 수 있는지 확인
    
    if len(user) != len(banned):
        return False
    
    for i in range(len(user)):
        if user[i] != banned[i] and banned[i] != '*':
            return False
            
    # user와 banned 아이디의 길이가 같고 모든 원소가 동일하거나 *인 부분일 경우 "같다"
    return True