프로그래머스 (2018년)KAKAO BLIND RECRUITMENT 오픈채팅방 c++

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

코딩테스트 연습 - 오픈채팅방 | 프로그래머스

이 문제는 map을 사용하여서 풀 수 있다. 각 유저 아이디를 key값으로 닉네임을 value로 map에 저장하면 된다.

나중에 닉네임을 바꾸거나 나갔다가 새로 들어올 때, 새로운 닉네임으로 적용돼야 하므로 해당 유저 아이디에 대한 닉네임을 바꿔줘야 한다. 즉, map에 이미 유저 아이디가 이미 존재하는 경우에는 새로운 닉네임(value)으로 바꿔버리면 된다.

 

 

사실 c++을 사용하다 보니까 map이 문제가 아니고 문자열 처리가 귀찮았다...

먼저 각 record에서 유저 아이디와 닉네임을 구분하기 위해서 공백을 기준으로 잘라줘야 한다.

C++에서 이 부분을 처리하려면 다른 방법도 있지만 char 배열과 strtok함수를 사용하면 된다.

strtok 함수는 csting을 include 해주면 된다.

 

 

1. 먼저 string을 char배열로 바꾼 후에

2. char 배열을 공백을 기준으로 잘라주고

3. 다시 string 2차원 배열에 넣어준다.

 

 

string 2차원 배열은 arr[100000][3]으로 선언해놨다. 100000은 문제에서 주어지는 record의 최대 길이이다.

 

arr[index][0] 은 index번째 record의 Enter/Change/Leave 중 하나의 값을 가지고

arr[index][1] 은 index번째 record의 유저 아이디

arr[index][2] 는 index번째 record의 닉네임을 가지도록 저장하였다.

 

모든 record에 대해서 공백을 기준으로 자르고 다시 새로운 string 배열에 넣어줬다면

각 record의 첫 번째 문자열이 Enter이냐 Leave이냐에 따라서 처리해주면 된다(Change일 때는 메시지가 안 뜨므로 처리해줄 필요 없다)

저장해둔 map을 이용하여 Enter인 경우에는 map[key]값과 함께 "님이 들어왔습니다"를 출력해주면 되고

Leave인 경우에는 map[key]값과 함께 "님이 나갔습니다"를 출력해주면 된다.

 

#include <string>
#include <vector>
#include <cstring>
#include <map>
 
using namespace std;
 
string arr[100000][3];
 
vector<string> solution(vector<string> record) {
    vector<string> answer;
    
    char str_buff[30];
    map<string, string> m;
    
    int index = 0;
    for(string r : record) {
        //string을 char array로 바꿔준다.
        strcpy(str_buff, r.c_str());
        int cnt = 0;
        
        //공백을 기준으로 자른다.
        char *tok = strtok(str_buff, " ");
        while (tok != nullptr) {
            //다시 string 배열에 넣어준다. 
            arr[index][cnt++] = string(tok);
            
            //다시 공백을 기준으로 자른다.
            tok = strtok(nullptr, " ");
        }
        
        
        //새로 들어오거나 이름을 변경한 경우에만 map에 추가 및 수정
        if(arr[index][0] == "Enter" || arr[index][0] == "Change") {
            if(m.find(arr[index][1]) == m.end()) {
                //map에 유저아이디가 없다면 새로 추가해준다.
                m.insert(make_pair(arr[index][1],arr[index][2]));
            } else {
                //map에 유저아이디가 이미 있으면 새로운 닉네임으로 바꿔준다.
                m[arr[index][1]] = arr[index][2];
            }
        }
        
        index++;
    }
    
    
    int size = record.size();
    for(int i=0; i<size; i++) {
        if(arr[i][0] == "Enter") {
            answer.push_back(m[arr[i][1]] + "님이 들어왔습니다.");
        } else if(arr[i][0] == "Leave") {
            answer.push_back(m[arr[i][1]] + "님이 나갔습니다.");
        }
    }
    
    
    
    return answer;
}