[프로그래머스 LV 2 ] 오픈채팅방 C++/python

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

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

 

c++로 sstream을 다뤄본 좋은 문제

문자열에대해 좀 나가보자

파이썬은 딕셔너리가 c++의 맵 처럼 갱신기능이 있다는걸알게됨

 

C++

#include <string>
#include <vector>
#include <sstream>
#include <unordered_map>
 
using namespace std;
 
vector<string> solution(vector<string> record) {
    vector<string> answer;
    stringstream ss;
    unordered_map <string, string> id_name;
    string e_l; string id; string name;
    vector<string> uid;
    for (int i = 0; i < record.size(); i++) {
        ss.str(record[i]);
        ss >> e_l;
        if (e_l == "Enter") {
            ss >> id;
            ss >> name;
            answer.push_back("님이 들어왔습니다.");
            id_name[id] = { name };
            uid.push_back(id);
        }
        else if (e_l == "Leave") {
            ss >> id;
            ss >> name;
            answer.push_back("님이 나갔습니다.");
            uid.push_back(id);
        }
        else {
            ss >> id;
            ss >> name;
            id_name[id] = { name };
        }
        ss.clear();
    }
 
    for (int i = 0; i < answer.size(); i++) {
        answer[i] = id_name[uid[i]] + answer[i];
    }
 
    return answer;
}

 

 

파이썬코드

c++에 있는 맵 대신 딕셔너리를 사용함

딕셔너리는 키가 중복되더라도 뒤에것을 쓰기때문에 갱신 기능이있음

def solution(record):
    answer = []
    uid = []
    dic = {}
 
    for i in record:
        if i.split(" ")[0] == "Enter":
            answer.append("님이 들어왔습니다.")
            dic[i.split(' ')[1]] = i.split(' ')[2]
            uid.append(i.split(' ')[1])
        elif i.split(' ')[0] == 'Leave':
            answer.append("님이 나갔습니다.")
            uid.append(i.split(' ')[1])
 
        else:
            dic[i.split(' ')[1]] = i.split(' ')[2]
 
    for i in range(len(answer)):
        answer[i]=dic[uid[i]]+answer[i]
    
    return answer