[프로그래머스 Level3] 불량 사용자 - 파이썬(Python) - 2019 카카오 개발자 겨울 인턴십

https://school.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 함수를 완성해주세요.

 

 

---

문제를 풀기 위해선, *가 들어있는 불량 아이디들 각각에 들어갈 수 있는 사용자 아이디들을 찾아야 한다.

다음은 불량 아이디들 각각에 들어갈 수 있는 사용자 아이디들을 찾는 코드이다.

불량 아이디마다 사용자 아이디 전체를 순회하면서 '*' 부분을 제외한 나머지 알파벳이 모두 같은지 확인한다.

# 사용자 아이디와 불량 사용자 아이디가 일치할 수 있는지 확인한다.
def is_sameID(id1, id2): 
    for i in range(len(id2)):
        if id2[i] != '*' and id1[i] != id2[i]:
            return False
    return True

banned_userID_list = [] # 불량 사용자 아이디 목록에 들어갈 수 있는 아이디들을 찾는다.
for bannedID in bannedID_list:
    userIDs = []
    for userID in userID_list:
        if len(userID)==len(bannedID) and is_sameID(userID, bannedID):
            userIDs.append(userID)
    banned_userID_list.append(userIDs)

 

 

이 뒤가 조금 헷갈렸는데,

예로 들어 불량 아이디가 3개가 있고 각각 가능한 아이디가 [2, 3, 1] 개 있다고 한다면, 순회가 인덱스 [0,0,0], [0,0,1], [0,1,0], [0,1,1], [0,2,0], [0,2,1], [1,0,0], ... [1,2,0]을 돌면서 그 3개의 선택된 아이디들이 같은게 있지는 않은지 계속 확인해봐야하나 싶은데 파이썬의 순열 모듈로는 접근할 수 없고 조금 헤맸다ㅠㅠ

 

여기서, 관점을 조금 바꿔서 유저 아이디들을 모두 순열로 순회하면서 각각이 불량 아이디가 가능한 아이디 목록마다 해당될 수 있는지 확인해보면 쉽게 풀릴 수 있었다..! 유저 아이디들을 전부 순열로 돌리면 중복도 없고 쉽게 제재 아이디 목록의 경우의 수를 구할 수 있었다.

from itertools import permutations

banned_cases = [] # 제재 아이디 목록의 경우의 수 구하기
for li in permutations(userID_list, len(banned_userID_list)):
    flag = True
    for i in range(len(li)):
        if li[i] not in banned_userID_list[i]:
            flag = False
            break
    if flag and set(li) not in banned_cases:
        answer += 1
        banned_cases.append(set(li))
return answer

 

 

<< 전체 코드 >>

from itertools import permutations

# 사용자 아이디와 불량 사용자 아이디가 일치할 수 있는지 확인한다.
def is_sameID(id1, id2): 
    for i in range(len(id2)):
        if id2[i] != '*' and id1[i] != id2[i]:
            return False
    return True
    
def solution(userID_list, bannedID_list):
    answer = 0
    banned_userID_list = [] # 불량 사용자 아이디 목록에 들어갈 수 있는 아이디들을 찾는다.
    for bannedID in bannedID_list:
        userIDs = []
        for userID in userID_list:
            if len(userID)==len(bannedID) and is_sameID(userID, bannedID):
                userIDs.append(userID)
        banned_userID_list.append(userIDs)
    
    banned_cases = [] # 제재 아이디 목록의 경우의 수 구하기
    for li in permutations(userID_list, len(banned_userID_list)):
        flag = True
        for i in range(len(li)):
            if li[i] not in banned_userID_list[i]:
                flag = False
                break
        if flag and set(li) not in banned_cases:
            answer += 1
            banned_cases.append(set(li))
    return answer