[프로그래머스 / Python파이썬] 불량 사용자

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

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

 

 

문제 설명

permutation을 이용해 user_id의 모든 가능한 순열을 구한 뒤 re(정규표현식)을 이용해 제재 아이디와 일치하는지를 확인했습니다. 

 

1. 순열 만들기

   - banned_id의 개수만큼 user_id의 원소를 뽑아서 만들 수 있는 순열을 생성합니다. 

 

2. 해당 순열로 제재 아이디 목록을 만들 수 있는지 확인

   - 문자열의 길이가 다르거나 (len(banned_id[i] != len(p[i])

   - 정규표현식에 맞지 않는 경우 (re.match(banned_id[i].replace('*', '.'), p[i]))

   - 순열의 i번째 원소가 banned_id의 i번째 원소와 매치되지 않습니다. 

   - 순열의 n개의 원소로 제재 목록의 n개의 id를 만들 수 있다면 answer에 해당 순열을 집합으로 바꿔서 추가합니다. 

 

   - 참고) 정규표현식에서 re.match는 문자열의 처음부터 정규식과 매치되는지 조사하여, 정규식에 부합하면 match 객체를 그렇지 않으면 None을 return 합니다. 

   - 참고) 정규 표현식의 Dot(.) 메타 문자는 줄 바꿈 문자인 \n을 제외한 모든 문자와 매치됨을 의미합니다. 

 

3. 가능한 경우의 수 구하기

   - 가능한 user_id의 조합(집합 자료형으로 저장했으므로 순열이 아닌 조합이 됨)이 answer에 저장되어 있으므로 answer의 길이를 return 합니다. 

   - answer 또한 집합이기 때문에 집합 내에 집합을 저장하기 위해서 frozenset을 이용했습니다. 

 

 

 

 

 

소스코드

from itertools import permutations
import re

def solution(user_id, banned_id):
    answer = set()
    n = len(banned_id) # 제재 아이디 개수
    perm = list(permutations(user_id, n)) # user_id의 n개 원소로 순열 생성

    for p in perm:
        cnt = 0
        # 아이디가 일치하는지 확인
        for i in range(n):
            if not re.match(banned_id[i].replace('*', '.'), p[i]) or len(banned_id[i]) != len(p[i]): 
                break
            else: 
                cnt += 1
        # p로 제재 아이디 목록은 만들 수 있음
        if cnt == n: 
            answer.add(frozenset(p))
    
    return len(answer)

 

 

 

참고 자료

점프 투 파이썬 - 정규표현식