[16165] 걸그룹 마스터 준석이

https://www.acmicpc.net/problem/16165

16165번: 걸그룹 마스터 준석이

<문제>

멤버 목록에 대한 쿼리에 쓸 <팀의 이름, {멤버 목록}>맵과

팀의 이름에 대한 쿼리에 쓸 <멤버 이름, 팀의 이름>맵을 구성하고, 멤버목록은 정렬시켜둔다.

 

<소스코드>

#include <bits/stdc++.h>
using namespace std;
int n, m;
map<string, vector<string>> mp;
map<string, string> grp;
int main(void) {
    cin >> n >> m;
    while (n--) {
        string name;
        int sz;
        cin >> name >> sz;
        while (sz--) {
            string member;
            cin >> member;
            mp[name].push_back(member);
            grp[member] = name;
        }
        sort(mp[name].begin(), mp[name].end());
    }
    while (m--) {
        string s;
        int key;
        cin >> s >> key;
        if (key == 0) {
            for (auto& i : mp[s]) cout << i << '\n';
        } else {
            cout << grp[s] << '\n';
        }
    }
    return 0;
}