회사에 있는 사람 //7785

김동완·2022년 7월 15일
0

BAEKJOON

목록 보기
3/53
  • 문제
    상근이는 세계적인 소프트웨어 회사 기글에서 일한다. 이 회사의 가장 큰 특징은 자유로운 출퇴근 시간이다. 따라서, 직원들은 반드시 9시부터 6시까지 회사에 있지 않아도 된다.
    각 직원은 자기가 원할 때 출근할 수 있고, 아무때나 퇴근할 수 있다.
    상근이는 모든 사람의 출입카드 시스템의 로그를 가지고 있다. 이 로그는 어떤 사람이 회사에 들어왔는지, 나갔는지가 기록되어져 있다. 로그가 주어졌을 때, 현재 회사에 있는 모든 사람을 구하는 프로그램을 작성하시오.
    //시간 제한: 1초, 메모리 제한: 256MB
  • 입력
    첫째 줄에 로그에 기록된 출입 기록의 수 n이 주어진다. (2 ≤ n ≤ 106) 다음 n개의 줄에는 출입 기록이 순서대로 주어지며, 각 사람의 이름이 주어지고 "enter"나 "leave"가 주어진다. "enter"인 경우는 출근, "leave"인 경우는 퇴근이다.
    회사에는 동명이인이 없으며, 대소문자가 다른 경우에는 다른 이름이다. 사람들의 이름은 알파벳 대소문자로 구성된 5글자 이하의 문자열이다.
  • 출력
    현재 회사에 있는 사람의 이름을 사전 순의 역순으로 한 줄에 한 명씩 출력한다.

// Created by dongwan-kim on 2022/07/10.
#include<iostream>
#include<algorithm>
#include<map>
using namespace std;

int n;
string name,check;		// 이름과, 출퇴근 받는 변수
map<string, string,greater<>> m;	//사전 순의 역순으로 출력이므로 greater<>
map<string, string>::iterator iter;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    cin>>n;
    for(int i=0;i<n;i++){
        cin>>name>>check;
        if(check=="enter"){		//출근이라면 map에 insert
            m.insert({name, check});
        }
        else{					//퇴근이라면 map에서 erase
            iter = m.find(name);
            m.erase(iter);
        }
    }
    for(iter=m.begin();iter!=m.end();iter++){	//회사에 남아있는 사람을 출력
        cout<<iter->first<<'\n';
    }
}

풀이

처음에는 vector를 사용하여 문제를 해결하려고 했다.

int main(){
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>name>>check;
        if(check=="enter"){
            v.push_back(name);
        }
        else{
            idx = find(v.begin(), v.end(), name)-v.begin();
            v.erase(v.begin()+idx);
        }
    }
    sort(v.begin(),v.end(),greater<>());
    for(int i=0;i<v.size();i++){
        cout<<v[i]<<"\n";
    }

}

정답코드와 구조는 완벽히 같고 vector와 map차이만 있다.
처음 vector로 풀었을 때 시간초과가 발생하였는데, 시간 초과가 발생할 부분은 find부분밖에 없기에 검색을 해보니 find함수가 n타임이라는 것을 알게 되었다.

find의 시간복잡도에 대해 검색하며 알아보다가 set과 map의 find시간복잡도가 O(log n)이라는 것을 알게되고 map stl에 대해 공부하고 문제를 해결했다.

profile
KIM DONGWAN

0개의 댓글