알고리즘/알고리즘 문제 풀이

(C++) 백준 7785번 - 회사에 있는 사람

용꿀 2023. 5. 22. 11:22

문제 링크 : https://www.acmicpc.net/problem/7785

 

7785번: 회사에 있는 사람

첫째 줄에 로그에 기록된 출입 기록의 수 n이 주어진다. (2 ≤ n ≤ 106) 다음 n개의 줄에는 출입 기록이 순서대로 주어지며, 각 사람의 이름이 주어지고 "enter"나 "leave"가 주어진다. "enter"인 경우는

www.acmicpc.net

문제 풀이

해시를 이용한 풀이이다.

unordered_set이라는 다소 생소한 STL을 사용하였다.

정렬은 cmp라는 cutom 비교 함수를 만들어서 사전순의 역순으로 정렬하였다.

#include <iostream>
#include <unordered_set>
#include <vector>
#include <algorithm>

using namespace std;

int n;
string person, act;
unordered_set<string> s;

bool cmp(string a, string b){
    return a>b;
}

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

    cin >> n;
    while(n--){
        cin >> person >> act;
        if(act == "enter") s.insert(person);
        else s.erase(person);
    }
    vector<string> v(s.begin(), s.end());
    sort(v.begin(), v.end(), cmp);
    for(auto k : v) cout << k << "\n";
}