[프로그래머스 Lv.3] 불량 사용자 (Python)

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/64064

프로그래머스

풀이

간단하게 *가 나오면 체크를 하지 않고, 그 외의 문자가 나오는 경우에는 같은 자리에서 문자가 일치하는지 확인하고 아니라면 체크를 종료하는 방식으로 풀이하였다.

처음에는 전체 회원 아이디 중 각 banned_id와 일치하는 경우를 뽑아놓고, 이 중에서 선택하여 조합을 만드려고 하였으나 구현이 복잡하여서 처음에 순열을 먼저 만들어두고 순열 전체를 체크하는 방식으로 변경하였다.

from itertools import permutations

def solution(user_id, banned_id):
    ups = list(permutations(user_id, len(banned_id))) # 전체 회원의 id 중 banned_id의 수만큼으로 이루어진 순열을 생성
    answer = []
    for up in ups: # 각 순열들을 순회하면서
        if check(banned_id, up): # 순열을 이루는 각 id가 각 banned_id의 패턴과 일치하는지 확인
            bid_set = set(up) # 일치하는 id들을 set으로 묶어 중복된 조합이 있는지 체크하도록 함
            if bid_set not in answer: # 중복된 것이 없다면
                answer.append(bid_set) # 추가
    return len(answer)

def check(bids, uids):
    for idx, uid in enumerate(uids): # 순열에서 각 위치의 user_id를 하나씩 꺼냄     
        target = bids[idx] # user_id와 동일한 위치의 banned_id
        if len(target) != len(uid): return False # 길이가 다르면 애초에 비교 대상이 아님

        for i in range(len(uid)): # 길이가 같다면 한 글자씩 비교
            if target[i] == "*": continue # "*"의 경우에는 체크하지 않음
            if target[i] != uid[i]: return False # 한 글자라도 다르면 비교 종료
    return True # 모든 패턴이 일치하면 True 반환