[20291] 파일 정리

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

20291번: 파일 정리

파싱은 find('.')이 return한 iterator의 다음것부터 취해주면 된다.

입력에서 확장자를 파싱한 다음에, multiset에 하나를 넣고 set에 하나를 넣는다고 생각하자.

기본적인 구현은 multiset으로 충분하나, 출력조건을 맞춰주기 위해서 중복을 허용/비허용하는 자료구조를 동시에 썼다.

---

#include <bits/stdc++.h>

using namespace std;

#ifdef ONLINE_JUDGE

constexpr bool local = false;

#else

constexpr bool local = true;

#endif

using ll = long long;

using pi = pair<ll, ll>;

int n;

multiset<string> st;

vector<string> v;

int main(void) {

    if (!local) ios_base::sync_with_stdio(0), cin.tie(0);

    cin >> n;

    int i;

    for (i = 0; i < n; i++) {

        string s;

        cin >> s;

        auto idx = find(s.begin(), s.end(), '.');

        s = string(++idx, s.end());

        st.insert(s);

        v.push_back(s);

    }

    sort(v.begin(), v.end());

    v.erase(unique(v.begin(), v.end()), v.end());

    int S = v.size();

    for (i = 0; i < S; i++) {

        string cur = v[i];

        cout << cur << ' ' << st.count(cur) << '\n';

    }

    return 0;

}

---