[프로그래머스] 신고 결과 받기 - 파이썬/python

문제

https://school.programmers.co.kr/learn/courses/30/lessons/92334

프로그래머스

불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 구현했을 때, 각 유저별로 처리 결과 메일을 받은 횟수를 구하는 문제다.

 

제한사항

• 2 ≤ id_list의 길이 ≤ 1,000
  • 1 ≤ id_list의 원소 길이 ≤ 10
  • id_list의 원소는 이용자의 id를 나타내는 문자열이며 알파벳 소문자로만 이루어져 있습니다.
  • id_list에는 같은 아이디가 중복해서 들어있지 않습니다.
• 1 ≤ report의 길이 ≤ 200,000
  • 3 ≤ report의 원소 길이 ≤ 21
  • report의 원소는 "이용자id 신고한id"형태의 문자열입니다.
  • 예를 들어 "muzi frodo"의 경우 "muzi"가 "frodo"를 신고했다는 의미입니다.
  • id는 알파벳 소문자로만 이루어져 있습니다.
  • 이용자id와 신고한id는 공백(스페이스)하나로 구분되어 있습니다.
  • 자기 자신을 신고하는 경우는 없습니다.
• 1 ≤ k ≤ 200, k는 자연수입니다.
• return 하는 배열은 id_list에 담긴 id 순서대로 각 유저가 받은 결과 메일 수를 담으면 됩니다.

 

예시

Example 1:

id_list: ["muzi", "frodo", "apeach", "neo"]
report: ["muzi frodo","apeach frodo","frodo neo","muzi neo","apeach muzi"]
k: 2
result: [2, 1, 1, 0]

 

Example 2:

id_list: ["con", "ryan"]
report: ["ryan con", "ryan con", "ryan con", "ryan con"]
k: 3
result: [0, 0]

 

풀이

모든 유저의 신고 기록을 저장하지 않고, 신고당한 사용자를 키로 하고, 그 사용자를 신고한 사람들의 리스트를 값으로 가지는 딕셔너리를 활용하면 문제를 효율적으로 해결할 수 있다.

def solution(id_list, report, k):
    answer = []
    
    # 동일한 유저에 대한 신고 1회로 처리
    report = set(report)
    
    # 신고당한 사람별 신고자 리스트
    reported = {}
    for i in report:
        a, r = i.split()
        if r not in reported:
            reported[r] = []
        reported[r].append(a)
    
    # 메일 받을 횟수 저장
    cnt = {user: 0 for user in id_list}
    
    # k 이상일 경우 신고자들에게 메일 전송
    for r, a in reported.items():
        if len(a) >= k:
            for j in a:
                cnt[j] += 1
        
    return [cnt[user] for user in id_list]

• 신고 기록 리스트에서 set을 사용하여 중복된 신고 기록을 제거한다. 
• 신고당한 사용자(r)를 키로, 해당 사용자를 신고한 사람들(a)을 값으로 가지는 딕셔너리를 만든다.
  • split()을 사용해 신고 기록에서 신고자와 신고당한 사람을 분리한다.
• 각 유저가 받을 메일의 횟수를 초기화한다.
  • 유저의 ID가 담긴 배열을 순회하며, 메일의 수를 담을 딕셔너리를 0으로 초기화한다.
• 정지 기준을 확인하여 메일 횟수를 증가시킨다.
  • 신고당한 사용자 딕셔너리를 순회하며, k번 이상 신고를 받은 유저를 찾아 신고한 사람들의 메일 횟수를 증가시킨다.
• 유저 ID 리스트를 순회하며, 각 유저가 받은 메일 수를 배열로 반환한다.

 

다른 사람의 풀이

훨씬 간결하고 직관적인 코드가 있어서 가져왔다.

def solution(id_list, report, k):
    answer = [0] * len(id_list)
    reports = {x: 0 for x in id_list}
    
    # 신고당한 사람 카운트 
    for r in set(report):
        reports[r.split()[1]] += 1
    
    # k 이상일 경우 신고자들 카운트
    for r in set(report):
        if reports[r.split()[1]] >= k:
            answer[id_list.index(r.split()[0])] += 1
        
    return answer

• 각 유저가 받는 메일의 횟수를 저장할 배열을 만든다. 초기값은 모두 0으로 설정한다.
• 신고당한 사람별 신고 횟수를 저장할 딕셔너리를 만든다. 각 유저는 0으로 초기화한다.
• set을 사용하여 중복 신고를 제거하고, 신고당한 사람의 신고 횟수를 계산한다.
• 다시 한 번 중복된 신고를 제거하고, 신고당한 사람의 횟수가 k 이상인 경우를 확인한다.
  • index를 사용하여 신고자의 인덱스를 찾아 해당 유저가 받는 메일을 하나 추가한다.