[ 프로그래머스 ] 신고 결과 받기 (C++) - 자료구조

---

🔗 문제 링크

프로그래머스

---

💡 문제 풀이 및 해석

1. 사용자마다 map<userId, set<userId>>를 사용해서 신고한 횟수가 아닌 신고 당한 ID를 모은다.
2. 신고 당한 ID의 갯수 = 나를 신고한 사람이므로, 이 점을 이용해서 풀면 된다.

---

⭐️ 정답 코드 및 설명

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

using namespace std;
map<string, set<string>> myReport; // 유저 : 해당 유저를 신고한 ID
map<string,int> user; // id : index

vector<int> solution(vector<string> id_list, vector<string> report, int k) {
    vector<int> answer;
    answer=vector<int>(id_list.size(), 0);

    int index=0;
    for(string id : id_list){
        user[id]=index++; // id : index
    }

    for(string str : report){
        string userId="",reportedId="";
        bool flag=false;
        for(char c : str){
            if(c==' '){
                flag=true;
                continue;
            }
            if(flag){
                reportedId+=c;
                continue;
            }
            userId+=c;
        }
        myReport[reportedId].insert(userId);
    }



    for(string id : id_list){
        if(myReport[id].size()>=k){
            for(string reportMe : myReport[id]){
                int i = user[reportMe];
                answer[i]++;
            }
        }
    }

    return answer;
}

---

🤔 문제 후기

map과 set을 이용하면 풀기에는 어렵지 않은 문제였지만, 그 중간에 Key : Value 에 대해서 실수만 하지 않으면 풀기에 어려운 문제는 아니였다. sstream을 사용하면, 쉽게 하는 방법이 있었는데, 정확하게 기억이 나지 않아서 flag를 사용해서 문자열을 잘라냈다.