[Programmers/파이썬]프로그래머스(Lv.3) 불량 사용 문제풀이

문제 설명

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

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

응모자 아이디

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_idbanned_idresult

["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

 

 

---

사고 흐름

불량 사용자 각각에 들어갈 수 있는 user_id를 찾자

-> 잠깐 단순하게 for문 4개 써서 할까? 생각도 했지만 가오 떨어져서 3개로 합의

-> 지금 내가 짠 코드는 급하게 짠거라서 훨씬 이쁘고 간결하게 수정할 수 있다..

-> 그렇게 각 불량 사용자에 들어갈 수 있는 user_id 리스트를 만들고

-> 그 리스트를 중복 없는 조합으로 만든다

-> 사실 이해하기가 좀 힘들었는데 그림을 그려보니 누가봐도 dfs라서 dfs로 풀이

-> 하지만 dfs 구현하는게 생각보다 오래걸려서 힘들었다.. 

-> 결국은 answer를 전역변수로 설정하고 방문한 리스트 길이가 충족되면 answer에 추가했다

-> 하지만 v.pop() 위치를 정하는게 또 당황스러웠따..ㅎ

 

문제 핵심과 알고리즘

문자열 비교를 통한 문자열 활용과 리스트 원소들의 조합 찾기(dfs 사용)

 

def solution(user_id, banned_id):
    global answer
    answer=[]
    ans = []

    for bid in banned_id:
        temp = []
        for uid in user_id:
            if len(bid) != len(uid):
                continue
            for (i, b) in enumerate(bid):
                if b == '*':
                    if i==len(bid)-1 :
                        temp.append(uid)
                    continue
                if bid[i] != uid[i]:
                    break
                if i==len(bid)-1 :
                    temp.append(uid)
        ans.append(temp)
        
    for i in range(len(ans[0])) :
        dfs(ans, ans[0][i], [], 1)
    
    return len(answer)

def dfs(ans, node, v, i) :
    global answer
    v.append(node)

    if i==len(ans) :
        new=sorted(v)
        if new not in answer :
            answer.append(new)
        return

    for a in ans[i] :
        if a not in v :
            dfs(ans, a, v, i+1)
            v.pop()

문제 풀면서 헤맸던 부분

1. dfs 구현

  dfs는 마스터했다고 생각했는데 완전 오산이였다. 재귀가 정말 싫다.. 진짜.. dfs 마스터가 되야겠다.. 외우는걸로 안되겠어ㅠ

 

다른 풀이 참고하여 배운 점

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)

이 풀이는 user_id와 banned_id를 돌면서 같은지 check 함수로 확인해줬고, 같다면 리스트에 추가했다. 그리고 나도 순열까지는 생각했지만 리스트가 여러개 있어서 사용하지 못했는데 product라는 아주 좋은,, 기능이 있었다. 그렇게 모든 경우의 수를 만들었고, set으로 중복 처리를 했다. 

=> product.. 아주 좋아

 

<개념 정리>

1. product

데카르트 곱이라고도 하는 cartesian product를 표현할 때 사용하는 메소드이다. 즉 두 개 이상의 리스트에서 모든 조합을 계산한다. 

from itertools import product

items = [['a','b'],['!','@'],[5,9]]
print(list(product(*items)))
#[('a', '!', 5), ('a', '!', 9), ('a', '@', 5), ('a', '@', 9), ('b', '!', 5), ('b', '!', 9), ('b', '@', 5), ('b', '@', 9)]

# 만약 중복이 존재한다면?
items = [['a','b'],['!','@'],['a','b']]
print(list(product(*items)))
#[('a', '!', 'a'), ('a', '!', 'b'), ('a', '@', 'a'), ('a', '@', 'b'), ('b', '!', 'a'), ('b', '!', 'b'), ('b', '@', 'a'), ('b', '@', 'b')]

#리스트 각 요소의 개수가 다르다면?
items = [['a'],['!'],[1,2,3,4,5,6,7,8,9]]
print(list(product(*items)))
#[('a', '!', 1), ('a', '!', 2), ('a', '!', 3), ('a', '!', 4), ('a', '!', 5), ('a', '!', 6), ('a', '!', 7), ('a', '!', 8), ('a', '!', 9)]