[c++] 프로그래머스 - 신고 결과 받기(Level 1)

문제

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

 

코드

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

using namespace std;

struct Info{
    set<string> member;
};

vector<int> solution(vector<string> id_list, vector<string> report, int k) {
    vector<int> answer;
    map<string,set<string>> db;
    map<string,int>reported_cnt;
    map<string,int>memory;
    
    for(string s: report){
        vector<string> report_split;
        istringstream ss(s);
        string stringBuffer;
        while(getline(ss, stringBuffer,' ')){   //공백 스플릿
            report_split.push_back(stringBuffer);
        }
        string reporter = report_split.front();
        
        for(int i=1; i<report_split.size(); i++){   // report 기록
            db[reporter].insert(report_split[i]);
        }
    }
    
    for(auto it=db.begin(); it!=db.end(); it++){    //신고 몇번 받았는지 기록
        for(auto it2 = it->second.begin(); it2!=it->second.end(); it2++){
            reported_cnt[*it2]++;
        }
    }
    
    for(auto it=db.begin(); it!=db.end(); it++){    //신고 몇번 받았는지 기록
        int cnt = 0;
        for(auto it2 = it->second.begin(); it2!=it->second.end(); it2++){
            if(reported_cnt[*it2]>=k) cnt++;
        }
        memory[it->first] = cnt;	// 해당 유저가 메세지를 몇개 받는지 기록
    }
    
    for(int i=0; i<id_list.size(); i++){
        answer.push_back(memory[id_list[i]]);	// 멤버 순서대로 메세지 갯수 저장
    }
    
    return answer;
}

 

풀이(45분)

입출력부터 공백을 split 처리를 해주어야 했기 때문에 C++로 풀기에 참 야박했던 문제였습니다.(물론 제 실력이 없는 것도 있겠지만)

• c++에서의 string split 기능을 배웠고,
• c++에서 set의 원소를 가져오기 위해서는 포인터로 접근해야 하는 것도 알게 되었던 문제였습니다.