문제
https://school.programmers.co.kr/learn/courses/30/lessons/92334
신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.
- 각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
- 신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
- 한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다.
- k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다.
- 유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다.
이용자의 ID가 담긴 문자열 배열 id_list, 각 이용자가 신고한 이용자의 ID 정보가 담긴 문자열 배열 report, 정지 기준이 되는 신고 횟수 k가 매개변수로 주어질 때, 각 유저별로 처리 결과 메일을 받은 횟수를 배열에 담아 return 하도록 solution 함수를 완성해주세요.
제한사항
- 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 순서대로 각 유저가 받은 결과 메일 수를 담으면 됩니다.
입출력 예
나의 풀이
from collections import defaultdict
def solution(id_list, report, k):
dict = defaultdict(set)
for r in report:
a, b = r.split()
dict[b].add(a)
answer = [0] * len(id_list)
for a in dict:
if len(dict[a]) >= k:
for b in dict[a]:
i = id_list.index(b)
answer[i] += 1
return answer각 유저가 누구에게 신고당했는지 저장하기 위해 딕셔너리를 사용했다.
- key: 신고당한 사람
- value: 신고한 사람들의 집합(set)
값은 중복 신고의 경우를 제외하기 위해 `set`을 사용했다.
신고 기록을 딕셔너리에 모두 추가한 후 k번 이상 신고된 유저를 찾아,
그 유저를 신고한 사람들에게 `answer`의 값을 1씩 증가시킨다.
다른 사람의 풀이
def solution(id_list, report, k):
reports = {x: 0 for x in id_list}
answer = {x: 0 for x in id_list}
for r in set(report):
a, b = r.split()
reports[b] += 1
for r in set(report):
a, b = r.split()
if reports[b] >= k:
answer[a] += 1
return list(answer.values())`set(report)`를 통해 탐색 대상에서부터 중복을 제거하고,
딕셔너리의 값에는 정수를 저장해서 신고 횟수를 카운트한다.
그리고 신고자별 메일 전송 횟수를 카운트하기 위해 미리 딕셔너리를 만들어서
`index()`가 아닌 딕셔너리 키 값을 통해 접근하도록 수정했다.