[Python] 프로그래머스: 신고 결과 받기

신고 결과 받기

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

코딩테스트 연습 - 신고 결과 받기

2022 KAKAO BLIND RECRUITMENT

문제 설명

신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.

• 각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
  • 신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
  • 한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다.
• k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다.
  • 유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다.

이용자의 ID가 담긴 문자열 배열 id_list, 각 이용자가 신고한 이용자의 ID 정보가 담긴 문자열 배열 report, 정지 기준이 되는 신고 횟수 k가 매개변수로 주어질 때, 각 유저별로 처리 결과 메일을 받은 횟수를 배열에 담아 return 하도록 solution 함수를 완성해주세요.

핵심 개념

dict, set 자료형을 이용한다.

풀이

def solution(id_list, report, k):
    '''
        id_list에 있는 사용자들에게 신고한 사용자에 대한 처리 결과가 몇 번 발송되었는지 구하여 return한다.
        
        Parameter:
        	id_list: 사용자의 id를 나타내는 str을 담은 list
            report:	사용자의 신고 내역을 나타내는 str을 담은 list
        	k: k번 이상 신고된 유저는 게시판 이용이 정지된다.
            
        Returns: id_list의 순서대로 사용자별로 몇 번 처리 결과가 발송되었는지를 담고 있는 list
    '''
    
    report_log = dict() # 사용자별 신고 사례를 담기 위한 dict
    user_mail = dict() # 사용자별 메일 발송 회수를 담기 위한 dict
    answer = []
    
    # 사용자 이름을 key, 그 사용자를 신고한 사용자를 담은 set를 value로 갖는 dict report_log를 작성
    for case in report:
        case = case.split() # 공백을 기준으로 분리
        if report_log.get(case[1]): # 신고당한 사용자의 key가 존재하면
            report_log[case[1]].add(case[0]) # 신고자를 value에 add
        else:
            report_log[case[1]] = {case[0]} # 중복 신고 사례는 세지 않으므로 set 자료형
    
    for user in id_list: # id_list에 있는 모든 유저에 대해서
        if report_log.get(user):
            if len(report_log[user]) >= k: # report_log에 key가 있고 신고한 사람이 k명 이상이면
                for reporter in report_log[user]: # 신고한 사용자들에게 메일 발송 회수가 +1
                    if user_mail.get(reporter):
                        user_mail[reporter] += 1
                    else:
                        user_mail[reporter] = 1
    
    for user in id_list:
        if user_mail.get(user):
            answer.append(user_mail[user])
        else: # user_mail에 key가 없으면, 한 번도 신고하지 않은 사용자
            answer.append(0)
    
    return answer