[Python] 프로그래머스 코딩테스트 연습 level 3 (불량 사용자)

1. 문제 설명

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

 

 

 

 

 

2. 제한 사항

• user_id 배열의 크기는 1 이상 8 이하입니다.
• user_id 배열 각 원소들의 값은 길이가 1 이상 8 이하인 문자열입니다.
  • 응모한 사용자 아이디들은 서로 중복되지 않습니다.
  • 응모한 사용자 아이디는 알파벳 소문자와 숫자로만으로 구성되어 있습니다.
• banned_id 배열의 크기는 1 이상 user_id 배열의 크기 이하입니다.
• banned_id 배열 각 원소들의 값은 길이가 1 이상 8 이하인 문자열입니다.
  • 불량 사용자 아이디는 알파벳 소문자와 숫자, 가리기 위한 문자 '*' 로만 이루어져 있습니다.
  • 불량 사용자 아이디는 '*' 문자를 하나 이상 포함하고 있습니다.
  • 불량 사용자 아이디 하나는 응모자 아이디 중 하나에 해당하고 같은 응모자 아이디가 중복해서 제재 아이디 목록에 들어가는 경우는 없습니다.
• 제재 아이디 목록들을 구했을 때 아이디들이 나열된 순서와 관계없이 아이디 목록의 내용이 동일하다면 같은 것으로 처리하여 하나로 세면 됩니다.

 

 

 

 

 

 

3. 내 풀이

from itertools import product
def solution(user_id, banned_id):
    answer = []
    for id in banned_id:
        star = []
        ids = []
        for i,x in enumerate(id):
            if x == '*':
                star.append(i)
        
        for u_id in user_id:
            if len(u_id) == len(id):
                new_id = u_id
                for s in star:
                    new_id = new_id[:s] + '*'+ new_id[s+1:]
                if new_id == id:
                    ids.append(u_id)
                    
        answer.append(ids)
    
    answer = list(product(*answer))
    answer = [sorted(x) for x in answer if len(x) == len(set(x))]
    answer = set(list(map(tuple,answer)))
    
    return len(answer)

 

1. for 문을 사용하여 제재 아이디들의 *의 위치를 리스트로 생성

2. 유저 ID 중 제재 ID와 길이가 같은 ID만 *의 위치 리스트를 이용하여 *로 변환함

3. 새로 생성된 유저 ID(*로 변환)이 제재 ID와 동일하면 리스트에 추가 -> 모든 제재 아이디들 반복

4. 최종적으로 각 제재 아이디 별 매핑된 유저아이디들이 담긴 리스트 생성

5. product() 함수를 사용하여 조합 생성 (answer)

6. 각 조합 중 중복된 요소가 있는 조합 제거

7. 최종 선택된 조합 중 중복 제거 (순서 상관없으므로 sorted()함수 사용해 정렬) 

8. 최종 리스트의 리스트 길이를 반환

 

사실 가독성이 많이 좋지는 않은 것 같아서 시간 여유될 때 더 고민해 봐야겠다...

 

 

 

4. 다른 사람의 풀이

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)

 

1. 아이디가 동일한지 여부를 반환하는 함수 생성

 1) 두 문자열의 길이가 다르면 False

 2) for 문을 사용하여 첫번째 문자열 중 '*'이 나오면 continue, 두 글자라 중 한글자라도 다르면 False 반환

2. answer을 set()으로 선언

3. 결과값을 받는 2차원 리스트를 banned_list의 길이만큼 생성

4. for문을 사용하여 제재 ID별 앞서 선언한 함수가 True이면 user_id를 result 리스트에 추가

5. for문을 사용하여 2차원 리스트 내에서 중복된 리스트 제거 후 리스트 길이 반환