문제 92334. 신고 결과 받기

문제 링크

https://school.programmers.co.kr/learn/courses/30/lessons/92334

프로그래머스

 

 

난이도 : Lv.1 

 

문제 요약 설명

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

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

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

입력

• ( 이용자의 ID가 담긴 문자열 배열 id_list
• 각 이용자가 신고한 이용자의 ID 정보가 담긴 문자열 배열 report
• 정지 기준이 되는 신고 횟수 k

입력 제한

• 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 순서대로 각 유저가 받은 결과 메일 수를 담으면 됩니다.

입력 예제

// input
["muzi", "frodo", "apeach", "neo"] // id_list
["muzi frodo","apeach frodo","frodo neo","muzi neo","apeach muzi"] // report
2 // k

// ans
[2,1,1,0]

 

풀이 방식

각 이용자가 신고한 이용자의 ID 정보가 담긴 문자열 배열 report를 이용하며 누가 누구를 신고했는지에 대한 정보를 담은  2차원 map자료구조인 m과 각 사용자의 신고 건수를 저장하는 m_cnt에 데이터를 저장한다. m과 m_cnt는 다음과 같이 생겼다.

 

ex) report: ["muzi frodo","apeach frodo","frodo neo","muzi neo","apeach muzi"] 

 

m 구조

	frodo	neo	muzi
muzi	true	true	-
apeach	true	-	true
frodo	-	true	-

map자료구조이기 때문에 실제로 배열처럼 생기지는 않았지만 이해를 위해 배열처럼 표현했다. m의 행은 신고를 보낸 유저(sender), 열은 신고받은 유저(receiver)를 의미한다.

 

m_cnt 구조

frodo	2
neo	2
muzi	1

 

m과 m_cnt에 데이터를 전부 저장했다면, id_list를 순회하며, 각 id를 가진 유저가 받을 메일 수를 계산한다.

유저가 받을 메일 수를 구하는 방법은 해당 유저가 신고한 신고받은 유저들(receiver)의 데이터를 담고있는 m의 열을 순회하며 각 receiver들이 신고받은 횟수를 m_cnt에서 찾는다. 만약, m_cnt에서 찾은 값이 k이상이라면 결과 메일 수를 1 증가시킨다.

이를 반복하여 모든 유저들이 받을 결과 메일 수를 계산하여 answer에 저장한다.

정답 코드 

#include <string>
#include <vector>
#include <map>
#include <iostream>
#include <sstream>

using namespace std;

vector<int> solution(vector<string> id_list, vector<string> report, int k) {
    vector<int> answer;
    map<string, map<string, bool>> m;   // report변수 2차원 배열로 재구성
    map<string, int> m_cnt;             // 각 사용자의 신고 받은 건수
    
    // 모든 신고 데이터를 저장
    string sender, receiver;
    for(string s : report){
        stringstream ss(s);
        ss >> sender;
        ss >> receiver;
        
        if(!m[sender][receiver]){
            m[sender][receiver] = true; 
            m_cnt[receiver]++;
        }
         
    }
    
    for(string s : id_list){
        int ban_cnt = 0;
        for(auto it : m[s]){
            if(m_cnt[it.first] >= k){
                ban_cnt++;
            } 
        }
        answer.push_back(ban_cnt);
    }
    
    return answer;
}