[Programmers] 신고 결과 받기

 

문제 

 

문제 설명

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

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

다음은 전체 유저 목록이 ["muzi", "frodo", "apeach", "neo"]이고, k = 2(즉, 2번 이상 신고당하면 이용 정지)인 경우의 예시입니다.

 

"muzi"	"frodo"	"muzi"가 "frodo"를 신고했습니다.
"apeach"	"frodo"	"apeach"가 "frodo"를 신고했습니다.
"frodo"	"neo"	"frodo"가 "neo"를 신고했습니다.
"muzi"	"neo"	"muzi"가 "neo"를 신고했습니다.
"apeach"	"muzi"	"apeach"가 "muzi"를 신고했습니다.

각 유저별로 신고당한 횟수는 다음과 같습니다.

 

"muzi"	1
"frodo"	2
"apeach"	0
"neo"	2

위 예시에서는 2번 이상 신고당한 "frodo"와 "neo"의 게시판 이용이 정지됩니다. 이때, 각 유저별로 신고한 아이디와 정지된 아이디를 정리하면 다음과 같습니다.

 

"muzi"	["frodo", "neo"]	["frodo", "neo"]
"frodo"	["neo"]	["neo"]
"apeach"	["muzi", "frodo"]	["frodo"]
"neo"	없음	없음

따라서 "muzi"는 처리 결과 메일을 2회, "frodo"와 "apeach"는 각각 처리 결과 메일을 1회 받게 됩니다.

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

 

 

나의 코드 

def solution(id_list, report, k):
    arr = []
    hashmap = {}
    answer = []
    imp = []
    cnt = 0
    
    # report 배열 이차원 배열로 만들기 (같은 사람이 같은 사람 신고한 중복 경우 제외)
    for rep in report:
        repo = rep.split(" ")
        arr.append(repo)
        if arr.count(repo) > 1:
            arr.pop()
    print(arr) 
    
    # hashmap으로 각 인물당 신고가 몇번됐는지 파악하기 
    for i in range(len(arr)):
        if arr[i][1] in hashmap:
            hashmap[arr[i][1]] += 1
        else:
            hashmap[arr[i][1]] = arr[i][1].count(arr[i][1])
    
    # 신고받은 횟수가 k 이상인 경우만 배열에 추가  
    for key in hashmap.keys():
        if hashmap[key] >= k:
            imp.append(key)
    print(imp)
    
    # 수정된 신고리스트를 바탕으로 메일 발송 횟수 구하기 
    for id in id_list:
        for t in range(len(arr)):
            if id == arr[t][0]:
                if arr[t][1] in imp:
                    cnt += 1
                else:
                    continue
        answer.append(cnt)
        cnt = 0
    return answer

분석 > 이 코드의 경우 report의 길이가 최대일 때 최대 400억번 돌기 때문에 시간초과가 뜬다. 그러기 때문에 브루트포스로 풀면 안된다는 것을 깨달었다. 

 

 

해설 코드 

def solution(id_list, report, k):
    
    reported_cnt = {x:0 for x in id_list}
    recieved_mail_cnt = [0] * len(id_list)
    
    for r in set(report):
        reported_cnt[r.split()[1]] += 1
    
    for r in set(report):
        if reported_cnt[r.split()[1]] >= k:
            recieved_mail_cnt[id_list.index(r.split()[0])] += 1
    return recieved_mail_cnt

분석 > 우선 신고 누적 횟수를 나타내는 해시맵을 미리 0으로 초기화하고 메일을 받는 횟수 또한 id_list를 기준으로 0으로 초기화 시킨다. 그리고 한 사람이 같은 사람을 여러번 신고한 것은 제외하기 위해 set(report)로 반복문을 돌리고 예를 들어 "muzi(신고한 사람) appeach(신고 받은 사람)" 이 r로 들어오면 각각 r.split()[0],r.split()[1] 로 나타내어 해시맵에 추가한다. 이어서 해시맵의 key 값이 되는 신고받은 사람(r.split()[1])으로 누적횟수가 k이상인 경우만 1을 증가시켜준다. 

 

 

 

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