신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.
유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다.
다음은 전체 유저 목록이 ["muzi", "frodo", "apeach", "neo"]이고, k = 2(즉, 2번 이상 신고당하면 이용 정지)인 경우의 예시입니다.
각 유저별로 신고당한 횟수는 다음과 같습니다.
위 예시에서는 2번 이상 신고당한 "frodo"와 "neo"의 게시판 이용이 정지됩니다. 이때, 각 유저별로 신고한 아이디와 정지된 아이디를 정리하면 다음과 같습니다.
def solution(id_list, report, k):
report = list(set(report))
ban= [[] for _ in range(len(id_list))] # 신고한 아이디들 id_list 순으로
banned = [0 for _ in range(len(id_list))] # 신고 당한 아이디들 횟수 id_list 순으로
result = [0] * len(id_list) # 신고한 아이디중 정지된 아이디
for id in report:
a = id.split(' ')
whereban = id_list.index(a[0]) #id_list 순서대로 저장하기 위해 신고한 아이디 인덱스 받아옴
ban[whereban].append(a[1])
wherebanned = id_list.index(a[1]) # 신고 당한 아이디 인덱스 받아옴
banned[wherebanned] += 1
for idx, id in enumerate(ban):
for i in id:
where = id_list.index(i)
if banned[where] >= k:
result[idx] += 1
return result
처음에 딕셔너리로 시도했다가 해결이 안돼서
인덱스 순서 맞춰주는 방법으로 구현했다.(복잡해진 감이 있음)
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
for r in set(report):
if reports[r.split()[1]] >= k:
answer[id_list.index(r.split()[0])] += 1
return answer
바로 딕셔너리로 만드는 방법
.split()해서 바로 슬라이싱