[프로그래머스] #42888 오픈채팅방

2018년 KAKAO BLIND RECRUITMENT

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

 

문제

채팅방에 들어오고 나가거나, 닉네임을 변경한 기록이 담긴 문자열 배열 record가 매개변수로 주어질 때, 모든 기록이 처리된 후, 최종적으로 방을 개설한 사람이 보게 되는 메시지를 문자열 배열 형태로 return 하도록 solution 함수를 완성하라.

풀이

문자열을 split 해야하는 부분에서 C++에는 해당 라이브러리가 없어서 문자열 다루는 부분에서 굉장히 애썼다.

strtok 를 활용하여 split과 같은 효과를 낼 수 있는데 자료형이 char* 이기 때문에 record 의 자료형인 string 을 먼저 char 형으로 바꾼 다음 포인터를 이용하여 문자열들을 구분했다. 

21, 27, 31 번 줄 같은 부분은

tok = strtok(NULL, " ");

while문 내부에 첫번째 파라메터로 NULL을 주게되면 내부 버퍼에 저장되어있는 포인터에서 다음 " "를 찾고 그 뒤에 문자열의 주소를 반환하게 된다. 이렇게 해서 계속해서 순회하면 모든 문자열을 잘라낼 수 있다.

문자열 처리를 마치고 나면 Enter 또는 Change 일 때는 세 덩이, Leave 일 때는 두 덩이로 나뉘게 된다. 이 때 마지막에 한 번에 문자열로 출력하기 위하여 모든 내용을 정리하여 vector v 에 저장해두는데 pair 쌍을 이용하여 첫 번째 명령어 부분이 Enter 이라면 second 값에 1을, Leave 라면 second 값에 0을 넣는다.

그리고 나서 v의 size 만큼 반복하여 차례로 second 값을 확인하여 1이라면 "님이 들어왔습니다.", 0이라면 "님이 나갔습니다."를 마지막 리턴될 배열인 answer에 저장해준다.

 

코드

#include <string>
#include <vector>
#include <map>
using namespace std;
 
vector<string> solution(vector<string> record) {
    typedef pair<string, int> p;
    vector<p> v;
    vector<string> answer;
    map<string, string> user;
    string uid, nickname;
    for (int i = 0; i < record.size(); i++) {
        vector<char> str(record[i].begin(), record[i].end());
        str.push_back('\0');
        char* p_str = &str[0];
        char* tok = strtok(p_str, " ");
        
        if (!strcmp(tok, "Enter")) {
            tok = strtok(NULL, " ");
            uid = tok;
            tok = strtok(NULL, " ");
            nickname = tok;
            user[uid] = nickname;
            v.push_back({ uid, 1 });
        }
        else if (!strcmp(tok, "Leave")) {
            tok = strtok(NULL, " ");
            uid = tok;
            v.push_back({ uid, 0 });
        }
        else { // Change
            tok = strtok(NULL, " ");
            uid = tok;
            tok = strtok(NULL, " ");
            nickname = tok;
            user[uid] = nickname;
        }
    }
    for (int i = 0; i < v.size(); i++) {
        if (v[i].second) {
            nickname = user[v[i].first];
            answer.push_back(nickname + "님이 들어왔습니다.");
        }
        else {
            nickname = user[v[i].first];
            answer.push_back(nickname + "님이 나갔습니다.");
        }
    }
    return answer;
}
 
int main() {
    solution({ "Enter uid1234 Muzi", "Enter uid4567 Prodo","Leave uid1234","Enter uid1234 Prodo","Change uid4567 Ryan" });
    return 0;
 
}

 

다른 사람 풀이 참고: sstream

 

#include <string>
#include <vector>
#include <sstream>
#include <iostream>
#include <map>
using namespace std;
 
 
vector<string> solution(vector<string> record) {
    vector<string> answer;
    string command;
    string ID;
    string uid;
   map<string,string> m;
 
 
    for(string input:record)
    {
        stringstream ss(input);
        ss>>command;
        ss>>uid;
        if(command=="Enter" || command=="Change")
        {
            ss>>ID;
            m[uid]=ID;
        }
    }
 
   for(string input:record)
    {
        stringstream ss(input);
        ss>>command;
        ss>>uid;
        if(command=="Enter")
        {
         ID=(m.find(uid)->second);
 
            string temp = ID+"님이 들어왔습니다.";
         answer.push_back(temp);
 
        }
      else if(command=="Leave")
      {
         ID=(m.find(uid)->second);
            string temp = ID+"님이 나갔습니다.";
         answer.push_back(temp);
      }
    }
    return answer;
}