[코딩 테스트 연습(파이썬/Python)] 프로그래머스 _ 불량 사용자

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

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

2019 카카오 인턴십 문제다.

[문제 설명]

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

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

응모자 아이디

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 이하인 문자열입니다.
  • 불량 사용자 아이디는 알파벳 소문자와 숫자, 가리기 위한 문자 '*' 로만 이루어져 있습니다.
  • 불량 사용자 아이디는 '*' 문자를 하나 이상 포함하고 있습니다.
  • 불량 사용자 아이디 하나는 응모자 아이디 중 하나에 해당하고 같은 응모자 아이디가 중복해서 제재 아이디 목록에 들어가는 경우는 없습니다.
• 제재 아이디 목록들을 구했을 때 아이디들이 나열된 순서와 관계없이 아이디 목록의 내용이 동일하다면 같은 것으로 처리하여 하나로 세면 됩니다.

[아이디어]

1) banned_id의 원소별 가능한 user_id의 원소를 사전으로 구성한 뒤, 그 value들로 가능한 모든 조합을 만들어 낸 뒤 중복을 제거하자.

2) product를 활용할 수 있을 것 같다.

 

# 정답
from itertools import permutations
from collections import Counter

def check(banned, user):
    if len(banned) != len(user):
        return False
    else:
        cnt = 0
        for i in range(len(banned)):
            if (banned[i] != user[i]) and (banned[i] == "*"):
                cnt +=1
            elif (banned[i] != user[i]) and (banned[i] != "*"):
                cnt -= 1
        if cnt == banned.count("*"):
            return True if cnt == banned.count("*") else False
        
def solution(user_id, banned_id):
    possible_banned = dict()
    for x in banned_id:
        for y in user_id:
            if check(x,y):
                possible_banned[x] = possible_banned.get(x, [])
                if y not in possible_banned[x]:
                    possible_banned[x] = possible_banned.get(x, []) + [y]
    a = list(permutations(range(len(user_id))))
    answer = []
    for x in a:
        tmp = banned_id.copy()
        tmp2 = []
        for y in x:
            if not check(tmp[-1], user_id[y]):
                break
            elif check(tmp[-1], user_id[y]):
                tmp.pop()
                tmp2.append(user_id[y])
            if not tmp:
                answer.append(tmp2)
                break
    
    return len(set(map(tuple, map(sorted, answer))))

산으로 가버린 풀이... 아이디어를 구상한대로 진행하지 못하고 방향을 틀었다. 가능한 모든 user_id의 순열을 만들어놓고 순서대로 뒤에서부터 banned_id와 매치하면 삭제를 진행했다. 이후 중복 제거,,

product를 활용하고자 할 때의 걸림돌이 banned_id에 중복이 존재하는 것과, 최종적으로도 중복 제거가 필요하다는 점이었는데 후자는 하다보니 tuple이 해결해줄 수 있는걸 깨달았고 전자는 아주 조금의 코딩을 추가해 해결한 코드가 있었다.

from itertools import product

def check(str1, str2):
    if len(str1) != len(str2):
        return False
    for i in range(len(str1)):
        if str1[i] == "*":
            continue
        if str1[i] != str2[i]:
            return False
    return True

def solution(user_id, banned_id):
    answer = set()
    result = [[] for i in range(len(banned_id))]

    for i in range(len(banned_id)):
        for u in user_id:
            if check(banned_id[i], u):
                result[i].append(u)

    result = list(product(*result))
    for r in result:
        if len(set(r)) == len(banned_id):
            answer.add("".join(sorted(set(r))))

    return len(answer)

갈 길이 멀구나,, 중간에 느꼈지만 dictionary에 또 너무 집착해서 list를 활용할 생각을 하지 못했다. (저번과 반대;) 애초에 주어진 배열의 최대 길이가 8이고, 꼭 dictionary를 활용할만큼 탐색이 잦은 문제도 아니었는데 저번 문제에 대한 기억으로 사전만 고집하다가 리스트 활용을 놓쳤다.. 걍 개수만큼 만들어놓고 넣어주면 될 것을,!

product를 활용하려면, 원하는 조합의 길이만큼의 iterator 개수를 원소로 갖는 배열을 만들어 준 뒤 product(*배열)을 해주면 된다. 위 사람처럼 길이로 확인하고 sorting 한 뒤 문자로 합쳐봐도 되고, 내가 한 것처럼 tuple을 활용해도 중복 제거가 가능하다.