Algorithm/문제풀이 Python

프로그래머스 레벨3 - 카카오 2019 인턴십 불량 사용자

nauni 2021. 1. 23. 22:33

문제주소

programmers.co.kr/learn/courses/30/lessons/64064?language=python3

내 풀이

import re
from itertools import  product

def solution(user_id, banned_id):
    answer_list = list()
    banned_list = []
    banned_length = len(banned_id)
    for banned in banned_id:
        banned = banned.replace("*", ".")
        p = re.compile(banned)
        id_list = []
        for user in user_id:
            if len(banned) == len(user) and p.match(user):
                id_list.append(user)
        banned_list.append(id_list)
    # 주의 : list(product(*banned_list))로 돌렸을 땐, 시간초과 나왔다.
    for candidate in product(*banned_list):
        candidate = set(candidate)
        if len(candidate) == banned_length and candidate not in answer_list:
            answer_list.append(candidate)
    return len(answer_list)

다른 사람 풀이

python으로 푼 풀이에는 거의 대부분 product 를 사용하였다. 정규표현식이 아니라, 함수를 하나 만들어서 길이가 같고,  * 일땐 넘기고, 아니면 같은지 체크하여 boolean으로 리턴해주는 함수로 같은 문자 리스트를 만들어준 사람들도 있었다. 

 

set도 sorted(set(candidate))의 형태로 정렬하여 문자열로 바꾼 뒤 set 자료형에 넣은 풀이도 있었다.  banned_list를 set으로 설정하고 중복되지 않게 set(candidate)을 sorting하여 문자열 형태로 banned_list(set자료구조)에 넣은 경우도 있었다.

정리

set안에 set 사용은 불가하다.

product 메소드로 리스트의 모든 조합을 구하기 가능하다.

set도 sorting이 가능하다.

* 가변인자 사용이 가능하다!! 리스트의 unpacking이 가능하다!!! ( 참고할만한 글1, 참고할만한 글2 )