[level 1] 신고 결과 받기

🔗 문제 링크

✔️ 문제 설명

신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다.
각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다.
k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다.
유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다.
다음은 전체 유저 목록이 ["muzi", "frodo", "apeach", "neo"]이고, k = 2(즉, 2번 이상 신고당하면 이용 정지)인 경우의 예시입니다.

유저 ID	유저가 신고한 ID	설명
"muzi"	"frodo"	"muzi"가 "frodo"를 신고했습니다.
"apeach"	"frodo"	"apeach"가 "frodo"를 신고했습니다.
"frodo"	"neo"	"frodo"가 "neo"를 신고했습니다.
"muzi"	"neo"	"muzi"가 "neo"를 신고했습니다.
"apeach"	"muzi"	"apeach"가 "muzi"를 신고했습니다.

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

유저 ID	신고당한 횟수
"muzi"	1
"frodo"	2
"apeach"	0
"neo"	2

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

유저 ID	유저가 신고한 ID	정지된 ID
"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 함수를 완성해주세요.

📚 코드

python

from collections import defaultdict

def solution(id_list, report_list, k):
    answer = [0 for _ in range(len(id_list))]
    
    report_count_dict = defaultdict(int)
    
    for report in set(report_list):
        reporter, reported = report.split(" ")
        report_count_dict[reported] += 1

    for report in set(report_list):
        reporter, reported = report.split(" ")
        if report_count_dict[reported] >= k:
            answer[id_list.index(reporter)] += 1
    
    return answer

Swift

import Foundation

func solution(_ id_list:[String], _ report:[String], _ k:Int) -> [Int] {
    
    var answer: [Int] = Array(repeating: 0, count: id_list.count)
    
    var reportedDict: [String: Int] = [:]
    var whoReportedDict: [String: Set<String>] = [:]
    
    var overReportedList: [String] = []
    
    Set(report).forEach {
        let reporterList = $0.components(separatedBy: " ")
        
        let reporter = reporterList[0]
        let reported = reporterList[1]
        
        whoReportedDict[reporter, default: []].insert(reported)
        reportedDict[reported, default: 0] += 1
    }
    
    for (reported, count) in reportedDict {
        if count >= k {
            overReportedList.append(reported)
        }
    }
    
    for (idx, id) in id_list.enumerated() {
        let reportList = whoReportedDict[id] ?? []
        
        overReportedList.forEach {
            if reportList.contains($0) {
                answer[idx] += 1
            }
        }
    }
    return answer
}

💡 풀이

• 먼저 파이썬으로 풀이할 때는 중복된 신고는 1회로 처리된다는 점을 활용하여 set을 사용했다. set을 사용하여 중복된 신고를 제거해주고 신고된 사람을 dict에 저장해주었다.
• 다시 한 번 report_list를 돌면서 만약 신고된 사람이 누적 신고횟수가 k번 이상인 경우에는 신고자의 index를 list에서 찾은 다음 answer 배열의 해당 index에 1을 더해주었다.
• 동일한 방법으로 swift 코드를 구현했으나 한 테스트 케이스에서만 시간 초과가 발생하여 다른 방식으로 풀이해보았다!
• 누가 몇 번 신고당했는지 저장하는 reportedDict, 누가 어떤 사람을 신고했는지 담고 있는 whoReportedDict, k번 이상 신고당한 사람의 리스트인 overReportedList
• 동일하게 신고된 사람의 누적 신고 횟수를 저장해주고 whoReportedDict에 신고한 사람:(신고당한 사람 리스트)를 저장해준다.
• reportedDict를 돌면서 만약에 k번 이상 신고당한 사람이 있다면 overReportedList에 추가해준다.
• id_list를 enumerate로 돌면서 그 사람이 신고한 리스트를 불러오고 다시 overReportedList를 돌면서 만약 over report된 사람이 그 사람이 신고한 리스트 내부에 있을 경우 answer[idx]에 1을 더해주었다.
• 너무 공간 복잡도가 커진 것 같아서 추후에 리팩토링을 진행해보고 싶은 코드,,