[2022 KAKAO BLIND RECRUITMENT] 신고결과받기

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

코딩테스트 연습 - 신고 결과 받기

 

string(id), set<string> (id 를 신고한 사람들)을 key - value 로 가지는 unordered_map 을 사용한다.

한 유저가 다른 유저를 2번 이상 신고한 것은 1번으로 처리하기 때문에 중복을 제거하는 set 을 사용한다.

 

report 배열은 띄워쓰기로 구분되어 정보가 들어온다. (신고한사람 신고받은사람)

그러므로  공백을 기준으로 문자열 파싱이 필요한데, 이는 stringstream 으로 쉽게 해결이 가능하다.

 

report 를 순회하며 문자열을 파싱해 신고한사람과 신고받은 사람의 id 를 분리한다

report_map[second].insert(first) 을 수행하는데, 이는

second 라는 이름의 id 에 해당하는 value 값인 set<string> 에 first 라는 이름의 id 가 중복을 허용하지 않고 inserting 된다는 것이다. 

 

이후, report_map 을 순회하여 set 의 size 가 k 보다 크거나 같은 경우에 대해서

set 을 순회하여, 해당하는 id 의 index 를 ++ 해주면된다.

이때 index 는 초기에 index 를 저장하는 unordered_map<string, int> id_map;  을 만들고

 

    for (int i = 0; i < id_list.size(); i++)
    {
        id_map.insert(make_pair(id_list[i], i));
    }

 

와 같이 초기화해주었다.

 

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

using namespace std;

vector<int> solution(vector<string> id_list, vector<string> report, int k) {
    vector<int> answer(id_list.size(), 0); // answer 초기화

    unordered_map<string, int> id_map;
    for (int i = 0; i < id_list.size(); i++)
    {
        id_map.insert(make_pair(id_list[i], i));
    }
    unordered_map<string, set<string>> report_map;
    for (auto ele : report)
    {
        stringstream ss(ele);
        string first, second;
        ss >> first, ss >> second;

        /*first 가 second 를 신고한 것*/
        report_map[second].insert(first);
        ss.clear();
    }

    for (auto ele : report_map)
    {
        if (ele.second.size() >= k)
        {
            for (auto iter : ele.second)
            {
                int index = id_map[iter];
                answer[index] ++;
            }
        }
    }

    return answer;
}

 

풀이 알고리즘이 잘 떠오르지 않아서, 풀이된 알고리즘을 여럿 참고하였다. 일단 많이 풀면서 감을 익혀보자.

 

배운 것

1. 문자열을 파싱해야 하는 경우에 stringstream 을 활용하면 쉽게 해결이 가능하다.

2. 생각보다 풀이 방법은 단순하다. 조금 더 고민해보자..

 

https://wadekang.tistory.com/6

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