[ Programmers / CodingTest / Python ] 불량 사용자

황승환·2022년 2월 21일
0

Python

목록 보기
192/498

문제 설명

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

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

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

[입출력 예]

user_id	banned_id	result
["frodo", "fradi", "crodo", "abc123", "frodoc"]	["fr*d*", "abc1**"]	2
["frodo", "fradi", "crodo", "abc123", "frodoc"]	["*rodo", "*rodo", "******"]	2
["frodo", "fradi", "crodo", "abc123", "frodoc"]	["fr*d*", "*rodo", "******", "******"]	3

입출력 예에 대한 설명

입출력 예 #1

문제 설명과 같습니다.

입출력 예 #2

다음과 같이 두 가지 경우가 있습니다.

제재 아이디
frodo
crodo
abc123
제재 아이디
frodo
crodo
frodoc

입출력 예 #3

다음과 같이 세 가지 경우가 있습니다.

제재 아이디
frodo
crodo
abc123
frodoc
제재 아이디
fradi
crodo
abc123
frodoc
제재 아이디
fradi
frodo
abc123
frodoc

접근 방법

이 문제는 파이썬의 permutations를 이용하였다. permutations를 만든 뒤에 각 permutations들을 banned_id와 비교하며 길이가 서로 다를 경우, False를 반환하도록 하고, 길이가 같을 경우 * 는 비교를 하지 않고 넘어가고 문자가 아예 다를 때에는 False를 반환하도록 하였다. 그리고 permuations를 모두 비교하고 난 뒤에는 True를 반환하도록 하였다.

True가 반환될 경우, 중복을 없애야 하므로 permuations를 set으로 변환한 뒤에 중복 체크를 걸쳐 정답을 모으는 리스트에 넣어주고 이 리스트의 길이를 최종적으로 반환하였다.

  • banned_id에 해당되는 여부를 확인해줄 함수 chk를 u, b를 인자로 갖도록 하여 선언한다.
    -> b의 길이만큼 반복하는 i에 대한 for문을 돌린다.
    --> 만약 u[i]의 길이가 b[i]의 길이와 같지 않을 경우,
    ---> False를 반환한다.
    --> u[i]의 길이만큼 반복하는 j에 대한 for문을 돌린다.
    ---> 만약 b[i][j]가 *일 경우, 다음 반복을 진행한다.
    ---> 만약 u[i][j]가 b[i][j]와 같지 않을 경우,
    ----> False를 반환한다.
    -> True를 반환한다.
  • user_id의 permuations를 구하여 u_per에 리스트 형태로 저장한다.
  • 정답을 담을 리스트 banned를 선언한다.
  • u_per을 순회하는 users에 대한 for문을 돌린다.
    -> 만약 chk(users, banned_id)가 False일 경우, 다음 반복으로 넘어간다.
    -> 그 외에는 users를 set()을 이용하여 집합으로 만든다.
    -> users가 banned에 없을 경우,
    --> users를 banned에 넣어준다.
  • banned의 길이를 반환한다.

solution.py

import itertools
def solution(user_id, banned_id):
    def chk(u, b):
        for i in range(len(b)):
            if len(u[i])!=len(b[i]):
                return False
            for j in range(len(u[i])):
                if b[i][j]=='*':
                    continue
                if b[i][j]!=u[i][j]:
                    return False
        return True
    u_per=list(itertools.permutations(user_id, len(banned_id)))
    banned=[]
    for users in u_per:
        if not chk(users, banned_id):
            continue
        else:
            users=set(users)
            if users not in banned:
                banned.append(users)
    return len(banned)

profile
꾸준함을 꿈꾸는 SW 전공 학부생의 개발 일기

0개의 댓글