[프로그래머스/PYTHON] 신고 결과 받기

프로그래머스 '신고 결과받기' 문제 풀이입니다.

 

문제 링크

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

● 문제 간단 설명

신고 Report를 통해 k번 이상 신고당한 유저를 찾고, 신고한 유저가 정지시킨 유저의 수를 배열로 return 하는 문제입니다.

 

● 문제 해결 방향

우선 각 report를 통해 유저들의 신고당한 횟수를 카운트하고,

신고당한 횟수가 k번이 넘는 유저의 경우, report에서 신고한 유저를 찾아 그 유저가 정지시킨 횟수를 카운트 해주었습니다.

 

 

● 문제 해결 코드

def solution(id_list, report, k):
    answer = [0] * len(id_list)
    report = set(report)
    
    reported = [ r.split(' ')[1] for r in report ]
    
    for id in id_list:
        if reported.count(id) >= k:
            for idIndex in range(len(id_list)):
                if id_list[idIndex] + ' ' + id in report:
                    answer[idIndex] += 1;
    
    return answer

위 코드도 통과되지만 중복 for문을 쓰는 부분이 마음에 걸렸고, 아래와 같이 로직을 변경할 수 있었습니다.

def solution(id_list, report, k):
    answer = [0] * len(id_list)
    report = set(report)
    
    reportCount = { id : 0 for id in id_list }
    reportedCount = { id : 0 for id in id_list }
    
    for r in report:
        reportedCount[r.split(' ')[1]] += 1
    
    for r in report:
        report, reported = r.split(' ')
        if reportedCount[reported] >= k:
            reportCount[report] += 1;
    
    return list(reportCount.values())

우선 신고자, 신고받은 사람을 각각 dict로 선언해주었고,

report를 통해 신고받은 사람의 신고받은 횟수를 카운트해주었습니다.

 

신고받은 카운트가 k번이 넘었을 경우에는 신고자의 정지시킨 횟수가 카운트되도록 해주었습니다.

 

● 참고

이번 문제는 제가 오랜만에 알고리즘 문제를 풀기도 하고...

짜고 보니 간단했지만, 문제 설명만 봤을 때는 뭔가 까다롭게 느껴지는 문제였습니다.