[프로그래머스 Lv2] : 오픈채팅방

문제 링크

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

코딩테스트 연습 - 오픈채팅방

오픈 채팅방의 출입 메세지를 리턴하는 문제.

풀이

사실 풀이 과정 자체는 어제 풀어봤던 1단계 문제들이 더 어려운 것 같다.

하지만 띄워쓰기가 포함된 문자열을 띄워쓰기를 기준으로 3개의 입력으로 나눠야한다는게 제일 까다로웠다

"Enter Uid1234 Prodo" 라고 입력이 들어오면

명령어인 "Enter"과 유저의 아이디인 "Uid1234" 그리고 닉네임 "Prodo"로 나눠야한다.

유저 아이디는 중복될 수 없지만 닉네임은 중복될 수 있다. 같은 Prodo라는 닉네임이라고 해도 2명의 유저일 수도 있다는 뜻이다.

• 입력문 쪼개기

처음에 나는 띄워쓰기가 몇 번째 인덱스인지 찾아서 자르는 방법을 사용했다.

pos1은 첫 번째 띄워쓰기의 위치, pos2는 두 번째 띄워쓰기의 위치라고 했을 때

0 ~ pos1앞까지 자르고 pos1 + 1 ~ pos2 앞까지 자르고 pos2 + 1 ~ 끝까지 이렇게 세 구간으로 자르는 방법을 썼다.

이때, pos1과 pos2는 find 함수를 이용해서 찾을 수 있다.

 

정답을 맞힌 후에 다른 사람의 코드를 한 번 봤더니 sstream을 이용하니까 훨씬 쉽게 입력을 받을 수 있었다.

        stringstream ss(record[i]);
        string cmd, ID, Nic;
        ss >> cmd;
        ss >> ID;
        ss >> Nic;

그냥 이렇게만 하면 record[i]에 있는 명령문을 띄워쓰기를 기준으로 3개를 받을 수 있다.

엄청 유용한 코드인 듯하다.

 

• 유저별로 닉네임 저장

유저가 닉네임을 변경하면 앞서 이미 생겨있던 출입메세지의 닉네임부분도 변경된다.

그렇기 때문에 유저별로 최종적으로 바뀐 닉네임을 알고 있어야한다.

유저의 아이디는 모두 고유한 정보이기 때문에 map을 이용했다.

유저의 아이디를 Key로 유저의 닉네임을 Value로 생각하면 된다.

map<string, string> User_Nickname이라는 맵을 선언했다.

 

• 메세지 출력

record 에 들어있는 명령문들 중에 Enter와 Leave인 경우에만 메세지를 출력하면 된다.

나는 출력해야할 메세지들의 정보를 간략하게 큐에 저장해두고 한 번에 출력하는 방식으로 했다.

queue<pair<char, string>> 을 해서 출력해야할 메세지가 Enter이면 'E'를, Leave이면 'L'을 페어의 첫번째에 넣고, 해당 유저의 ID를 페어의 두번째에 넣었다.

그런데 굳이 이렇게 안하고 record를 다시 돌면서 명령어가 Enter 또는 Leave인 경우에만 answer벡터에 메세지를 집어넣어도 될 것같다.

이렇게 하면 굳이 큐를 만들지 않아도 된다.

 

#include <string>
#include <vector>
#include <iostream>
#include <map>
#include <queue>
#include <sstream>
using namespace std;
const int MAX_N = 100000 + 5;
queue<pair<char, string>> q;
 
vector<string> solution(vector<string> record) {
    vector<string> answer;
    map<string, string> User_NickName;
    for (int i = 0; i < record.size(); i++) {
        stringstream ss(record[i]);
        string cmd, ID, Nic;
        ss >> cmd;
        ss >> ID;
        ss >> Nic;
        if (cmd == "Enter") {
            User_NickName[ID] = Nic;
            q.push({ 'E',ID });
        }
        else if (cmd == "Change")
            User_NickName[ID] = Nic;
        else if (cmd == "Leave") {
            q.push({ 'L',ID });
        }
    }
    while (!q.empty()) {
        char cmd = q.front().first;
        string ID = q.front().second;
        q.pop();
        string temp = "";
        temp += User_NickName[ID];
        temp += "님이 ";
        if (cmd == 'E')
            temp += "들어왔습니다.";
        else if (cmd == 'L')
            temp += "나갔습니다.";
        answer.push_back(temp);
    }
    return answer;
}