[프로그래머스 Lv.1] 신고 결과 받기 / c++

[출처] https://school.programmers.co.kr/learn/courses/30/lessons/92334

프로그래머스

 

 

▷▷ 문제

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

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

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

 

---

 

▶▶ KEY

1. 자료구조 설정 

• map<string, int> id_index : id_list의 string id와 int index를 저장
• map<string, vector<int>> report_map : 신고받은 receiver의 id와 그 receiver를 신고한 사람들 id vector를 저장

2. 문제 풀이

• 중복 신고 가능하나, 1회만 유효 → report에서 중복 제거
• report 읽으면서 report_map 완성하기
• report_map에서 신고자 vector의 size가 K 이상인 receiver는 정지됨
•   → 그 vector의 신고자들 index의 result값을 ++

 

---

▶▶ 답안

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

using namespace std;

vector<int> solution(vector<string> id_list, vector<string> report, int k) {
    vector<int> answer;
    map<string, int> id_index;
    map<string, vector<int>> report_map;

    
    // report 중복 제거
    sort(report.begin(), report.end());
    report.erase(unique(report.begin(), report.end()), report.end());
    
    
    // report_map 뼈대 생성
    for(int i=0 ; i<id_list.size() ; i++) {
        vector<int> names;
        report_map.insert(make_pair(id_list[i], names));
        
        id_index.insert(make_pair(id_list[i], i)); // id_index Map
        answer.push_back(0);
    }
    
    
    // report_map insert
    string sender, receiver;
    for(int i=0 ; i<report.size() ; i++) {
        stringstream ss(report[i]);
        ss.str(report[i]);
        
        ss >> sender;
        ss >> receiver;
        
        report_map[receiver].push_back(id_index[sender]);
    }
    
    
    // report_map 채워넣기
    map<string, vector<int>>::iterator iter;
    for(iter = report_map.begin() ; iter != report_map.end() ; iter++) {
        if(iter->second.size() >= k) {
            for(int i=0 ; i<iter->second.size() ; i++) {
                answer[iter->second[i]]++;
            }           
        }  
    }
    
    
    return answer;
}