[프로그래머스][Level1] 신고 결과 받기 c++

문제 링크 https://programmers.co.kr/learn/courses/30/lessons/92334

문제

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

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

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

풀이 과정

풀이 (시간초과 1개)

#include <string>
#include <vector>
#include <map>

using namespace std;

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

    map<pair<string, string>, int> repo; // report
    map<string, int> s; // 신고받은 유저

    for(int i = 0; i < report.size(); i++){
        int idx = report[i].find(' ');
        string from = report[i].substr(0, idx);
        string to = report[i].substr(idx);

        repo[{from, to}]++;
        if(repo[{from, to}] == 1) s[to]++;
    }
    vector<string> v; // 정지받은 유저
    for(auto iter : s){
        if(iter.second >= k) v.push_back(iter.first);
    }

    map<string, int> ans;
    for(int i = 0; i < v.size(); i++){
        for(auto iter : repo){
            if(iter.first.second == v[i]){
                ans[iter.first.first]++;
            }
        }
    }
    for(int i = 0; i < id_list.size(); i++){
        int cnt = 0;
        for(auto iter : ans){
            if(id_list[i] == iter.first){
                answer.push_back(iter.second);
                cnt++;
            }
        }
        if(!cnt) answer.push_back(0);
    }
    return answer;
}

풀이 (시간초과 2개)

#include <string>
#include <vector>
#include <map>

using namespace std;

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

    vector<vector<string>> repo;
    for(int i = 0; i < report.size(); i++){
        string str = "";
        vector<string> temp;
        for(int j = 0; j < report[i].size(); j++){
            if(report[i][j] == ' '){
                temp.push_back(str);
                str = "";
            }
            else str += report[i][j];
        }
        if(str != "") temp.push_back(str);
        repo.push_back(temp);
    }

    map<pair<string, string>, int> m; // m -> f, 1
    map<string, int> s; // 신고 받은 횟수
    for(int i = 0; i < repo.size(); i++){
        m[{repo[i][0], repo[i][1]}]++;
        if(m[{repo[i][0], repo[i][1]}] == 1){
            s[repo[i][1]]++;
        }
    }

    vector<string> v; // 정지된 유저
    for(auto i : s){
        if(i.second >= k)  v.push_back(i.first);
    }

    for(int i = 0; i < id_list.size(); i++){
        int cnt = 0;
        for(auto iter : m){
            if(id_list[i] == iter.first.first){
                for(int j = 0; j < v.size(); j++){
                    if(v[j] == iter.first.second){
                        cnt++;
                    }
                }
            }
        }
        answer.push_back(cnt);
    }
    return answer;
}