Programmers / Level 1 / 신고 결과 받기 / C++ / JS

문제 설명

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

• 각 유저는 한 번에 한 명의 유저를 신고할 수 있습니다.
  • 신고 횟수에 제한은 없습니다. 서로 다른 유저를 계속해서 신고할 수 있습니다.
  • 한 유저를 여러 번 신고할 수도 있지만, 동일한 유저에 대한 신고 횟수는 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 함수를 완성해주세요.

---

제한사항

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

---

입출력 예

 

id_list	report	k	result
["muzi", "frodo", "apeach", "neo"]	["muzi frodo","apeach frodo","frodo neo","muzi neo","apeach muzi"]	2	[2,1,1,0]
["con", "ryan"]	["ryan con", "ryan con", "ryan con", "ryan con"]	3	[0,0]

---

입출력 예 설명

입출력 예 #1

문제의 예시와 같습니다.

입출력 예 #2

"ryan"이 "con"을 4번 신고했으나, 주어진 조건에 따라 한 유저가 같은 유저를 여러 번 신고한 경우는 신고 횟수 1회로
처리합니다. 따라서 "con"은 1회 신고당했습니다. 3번 이상 신고당한 이용자는 없으며,
"con"과 "ryan"은 결과 메일을 받지 않습니다. 따라서 [0, 0]을 return 합니다.

 

 

<카카오 공채 페이지에 나와있는 풀이>

문제 풀이

이 문제는 해시 자료구조를 활용할 수 있는지 묻는 문제였습니다.

report를 하나씩 처리하면서 각 유저가 누구에게 신고를 당했는지 목록을 만듭니다.

[신고된 유저 아이디 1] : [신고한 유저 A, 신고한 유저 B, ... ]
[신고된 유저 아이디 2] : [신고한 유저 A, 신고한 유저 C, ... ]
...
[신고된 유저 아이디 N] : [신고한 유저 X, 신고한 유저 Y, ... ]

유저 아이디는 최대 1,000개, report의 길이는 최대 200,000이므로 신고된 유저 아이디를 배열에 관리한다면
최악의 경우 O(아이디 개수 * report의 길이) 만큼의 시간이 걸립니다.
따라서 신고된 유저 아이디를 해시(또는 연관 배열) 자료구조를 이용해 관리하면 효율적으로 목록을 완성할 수 있습니다. 이때, 신고한 유저 목록에는 같은 아이디가 중복되어 들어가지 않도록 주의해야 합니다.

위와 같이 목록을 만들었다면 신고된 유저 아이디를 순회하면서 정지 기준을 만족하는 유저가 있다면
(신고한 유저 수가 k 이상이면) 신고한 유저 목록을 순회하면서 메일을 보냈다는 표시(카운트를 1 증가) 해주면 됩니다.

 

 

<풀이 코드>

#include <string>
#include <vector>
#include <unordered_map>
#include <set>
#include <sstream>

using namespace std;

vector<int> solution(vector<string> id_list, vector<string> report, int k) {
    int size = id_list.size();
    vector<int> answer(size, 0);
    
    // id의 index를 저장
    unordered_map<string, int> id_idx;
    for(int i=0; i<size; i++) {
        id_idx.insert(make_pair(id_list[i], i));
    }
    
    // 각 id를 신고한 id 저장
    // unordered_map<신고당한id, set<신고한id>> report_map
    unordered_map<string, set<string>> report_map;
    for(auto const &element : report) {
        stringstream ss(element);
        // 신고한 id, 신고당한 id
        string id, reported_id;
        ss >> id >> reported_id;
        
        // 각각의 신고당한 id에 누가 신고했는지 insert 해준다.
        report_map[reported_id].insert(id);
    }
    
    for(auto it : report_map) {
        // 신고한 id가 k명 이상이면
        if(it.second.size() >= k) {
            for(auto report_id : it.second) {
                // 신고한 id에 해당되는 id의 index를 찾는다.
                int idx = id_idx[report_id];
                answer[idx]++;
            }
        }
    }
    
    return answer;
}

---

생각보다 헷갈리는 문제였다.

검색해서 여러 코드를 찾아보고 참고해서 코드를 짰다.

 

vector를 이용하여 unique로 중복을 제거해서 푼 코드들도 많았지만, 카카오의 공식 해설에 가장 어울리는 코드를

참고했다.

 

코드에서 여러 vector와 map을 참조하게 되는데, 헷갈리지 않고 index를 잘 참조하는게 중요한 것 같다.

map의 특성상 중복을 허용하지 않기때문에 따로 중복처리를 해줄필요가 없다.

---

< JavaScript >

function solution(id_list, report, k) {
  // Set()으로 중복 제거
  const report_info = [...new Set(report)].map(el => el.split(' '));

  // 신고당한 횟수 저장
  const reported_cnt = new Map();
  for (const report of report_info) {
    // 신고당했으면 cnt+1 해주고, 값이 초기화 되어 있지 않으면 1로 초기화
    reported_cnt.set(report[1], reported_cnt.get(report[1]) + 1 || 1);
  }

  // k번 이상 신고당한 id 저장
  const received = new Map();
  for (const report of report_info) {
    // 신고당한 횟수가 k 이상이면,
    if (reported_cnt.get(report[1]) >= k) {
      // 신고자 cnt+1
      received.set(report[0], received.get(report[0]) + 1 || 1);
    }
  }

  const result = id_list.map((el) => received.get(el) || 0);
  return result;
}