[Python] 신고 결과 받기 (프로그래머스 카카오 문제)

문제를 보시려면 클릭하세요

문제 설명

신입사원 무지는 게시판 불량 이용자를 신고하고 처리 결과를 메일로 발송하는 시스템을 개발하려 합니다. 무지가 개발하려는 시스템은 다음과 같습니다. 각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다. 신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다. 한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 1회로 처리됩니다. k번 이상 신고된 유저는 게시판 이용이 정지되며, 해당 유저를 신고한 모든 유저에게 정지 사실을 메일로 발송합니다. 유저가 신고한 모든 내용을 취합하여 마지막에 한꺼번에 게시판 이용 정지를 시키면서 정지 메일을 발송합니다. 다음은 전체 유저 목록이 ["muzi", "frodo", "apeach", "neo"]이고, k = 2(즉, 2번 이상 신고당하면 이용 정지)인 경우의 예시입니다. 각 유저별로 신고당한 횟수는 다음과 같습니다. 위 예시에서는 2번 이상 신고당한 "frodo"와 "neo"의 게시판 이용이 정지됩니다. 이때, 각 유저별로 신고한 아이디와 정지된 아이디를 정리하면 다음과 같습니다. 따라서 "muzi"는 처리 결과 메일을 2회, "frodo"와 "apeach"는 각각 처리 결과 메일을 1회 받게 됩니다. 이용자의 ID가 담긴 문자열 배열 id_list, 각 이용자가 신고한 이용자의 ID 정보가 담긴 문자열 배열 report, 정지 기준이 되는 신고 횟수 k가 매개변수로 주어질 때, 각 유저별로 처리 결과 메일을 받은 횟수를 배열에 담아 return 하도록 solution 함수를 완성해주세요.

제한 사항

2 ≤ id_list의 길이 ≤ 1,000 1 ≤ id_list의 원소 길이 ≤ 10 id_list의 원소는 이용자의 id를 나타내는 문자열이며 알파벳 소문자로만 이루어져 있습니다. id_list에는 같은 아이디가 중복해서 들어있지 않습니다. 1 ≤ report의 길이 ≤ 200,000 3 ≤ report의 원소 길이 ≤ 21 report의 원소는 "이용자id 신고한id"형태의 문자열입니다. 예를 들어 "muzi frodo"의 경우 "muzi"가 "frodo"를 신고했다는 의미입니다. id는 알파벳 소문자로만 이루어져 있습니다. 이용자id와 신고한id는 공백(스페이스)하나로 구분되어 있습니다. 자기 자신을 신고하는 경우는 없습니다. 1 ≤ k ≤ 200, k는 자연수입니다. return 하는 배열은 id_list에 담긴 id 순서대로 각 유저가 받은 결과 메일 수를 담으면 됩니다.

입출력 예

---

사담

- 이 문제는 시간초과 때문에 고민하고 로직을 수정하며 풀었습니다. 여러분들도 많은 고민과 시도 해보시는 것을 추천드립니다!

나의 코드

def solution(id_list, reports, k):
    stop = []
    answer = [0] * len(id_list)
    dicReports = {id: [] for id in id_list}
    for i in set(reports):
        report = i.split(' ')
        stop.append(report[1])
        dicReports[report[0]].append(report[1])
        
    stop = set([i for i in stop if stop.count(i) >= k])

    for key, value in dicReports.items():
        for s in stop:
            if s in value:
                answer[id_list.index(key)] += 1
    return answer

---

문제 설명

1) 기본 설명

- 프로그래머스에서 제공하는 기본 함수는 다음과 같습니다.

def solution(id_list, report, k):
    answer = []
    return answer

---

2) 아이디어

(1)  Dictionary를 사용하자

    -> 신고를 k번 이상 받았을 경우 정지됩니다. 정지 된 경우, 그 사람을 신고한 사람에게 메일이 가게 됩니다.

    -> 두 값을 따로 처리해야하지만, 서로 연관되어있음을 알 수 있습니다. (한 명이 여러 값을 가지는 상황이므로, Dictionary를 사용했습니다)

(2) 신고를 k번 받은 사람만 정지된다 -> key = 신고 받은 사람으로 설정한다.

    -> 이게 제 핵심 아이디어자, 시간 초과 문제를 해결할 수 있었던 방법입니다.

    -> 신고 받은 사람을 key로 두게 되면, value는 신고를 한 사람이 됩니다.

    -> value의 길이가 k번 이상이라면, 메일이 가게 됩니다.

---

3) 코드 설명

    dic_report = {id: [] for id in id_list}
    answer = [0] * len(id_list)
    for report in set(reports):
        report = report.split(' ')
        dic_report[report[1]].append(report[0])

- id_list를 통해 dictionary를 생성합니다.

- answer 리스트는 결과 값을 담을 리스트입니다. (메일 받을 횟수)

- report를 공백으로 구분하면, 첫번째에는 신고 한 사람, 두번째 인덱스에는 신고 받는 사람이 담깁니다.

    -> 신고 받은 사람의 value에 신고 한 사람을 넣습니다.

    -> value의 길이를 보면, 몇 번 신고당했는지 알 수 있게 됩니다.

---

    for key, value in dic_report.items():
        if len(value) >= k:
            for v in value:
                answer[id_list.index(v)] += 1

- 신고 정보가 담긴 Dictionary를 탐색합니다.

- 신고를 k번 이상 받은 경우

    -> 신고 한 사람 위치에 1을 더해줍니다.

        -> index함수를 이용하여 신고 한 사람의 위치를 찾아 넣어주는 것입니다.

4) 결과

---

➕시간초과 났던 코드 공유합니다. 궁금하시면 클릭해주세요!

def solution(id_list, reports, k):
    stop = [] # 신고받은 사람
    answer = [0] * len(id_list)
    dicReports = {id: [] for id in id_list}
    for i in set(reports):
        report = i.split(' ')
        stop.append(report[1]) # value = 신고 받은 사람
        dicReports[report[0]].append(report[1])

    stop = set([i for i in stop if stop.count(i) >= k]) # k번 이상 신고 받은 사람만 남음
    # 예시 1번에서 ["frodo", "neo"]가 들어감
    
    for key, value in dicReports.items():
        for s in stop: # stop 목록에 존재한다면
            if s in value:
                answer[id_list.index(key)] += 1 # answer에 삽입
    return answer

---

➕제 깃허브 들어오시면, 설명은 없지만 다른 문제들의 코드를 볼 수 있습니다.

GitHub - JIWON1923/Algorithm: studying algorithm with python

➕더 좋은 방법이나 개선점이 있으면 알려주세용🍎