Lv.1 : 신고 결과 받기

👉 문제링크

프로그래머스

---

---

🔸 문제 분석 🔸

• 신고받아 정지된 유저가 있으면, 신고한 사람에게 알람이간다.
• 유저들이 받은 알람 수 리스트를 반환한다.

🔸 코드 🔸

def solution(id_list, report, k):
    answer = {}
    for i in id_list:           # 유저 ID 명단
        answer[i] = 0
    
    dic = {}
    for s in report:            # 신고 현황 (중복제거)
        a, b = s.split()
        if b in dic.keys():
            dic[b].add(a)
        else:
            dic[b] = set()
            dic[b].add(a)

    for value in dic.values():  # 알람 카운트
        if len(value) >= k:
            for i in value:
                answer[i] += 1

    return list(answer.values())

🔸 코드 해석 🔸

• 유저 ID명단으로 answer 딕셔너리를 생성한다.
• 신고 현황을 dic 딕셔너리에 세트로 저장하며 중복을 제거한다.
  • 입력은 신고자:신고 받은 유저 형식이지만, 신고 받은 유저 : 신고자 형태로 바꾼다.
• dic 딕셔너리에서 values의 크기가 k 이상이면 answer의 값을 1 증가시킨다.

 

🔸 다른 풀이 🔸

from collections import defaultdict, Counter

def solution(id_list, report, k):

    report_dict = defaultdict(set) 
    report_cntr = Counter()

    # {user1가 신고한 user2들} 인 dict 생성
    for r in report: 
        user1, user2 = r.split()
        report_dict[user1].add(user2)

    # 신고당한횟수 인 dict 생성
    for v in report_dict.values():
        report_cntr.update(v)

    # {신고를 k번 이상 당한 사람들} set 생성
    blacklist = {x for x in report_cntr if report_cntr[x] >= k} 

    # 1과 3을 이용하여 user1이 신고한 사람들중에 k번 이상 신고당한 사람들 숫자 도출
    return [len(report_dict[user] & blacklist) for user in id_list]

 

• 신고 현황을 저장한 딕셔너리를 defaultdict(set)를 이용해 set 형식으로 생성한다.
• set형식 딕셔너리에 값을 넣으며 중복을 제거한다.
• Counter() 딕셔너리에 값을 넣고, k번 이상 신고된 블랙리스트 유저의 ID 리스트를 만든다.
• 처음 set형식 딕셔너리와 만들어진 블랙리스트 유저를 비교해 겹치는 원소의 개수를 리스트로 만들어 반환한다.

---

🔸 end 🔸

• 딕셔너리와 세트 사용을 연습해볼 수 있었던 문제였다.
• 다른 풀이에서 defalutdict와 Counter 사용을 다시금 배울 수 있었다.
  • &로 두 리스트 원소를 비교해 개수를 파악하는게 인상적이었다.
  • 반복문 수가 적어서 그런지 좀 더 빠른 처리가 된 테스트들이 있었다.