[C/C++] 백준 20920번 - 영단어 암기는 괴로워

✏️  문제 링크

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

 

✏️ 문제 설명

✏️ 로직

1. map<string,int>mp를 생성해서 단어와, 해당 단어의 개수를 저장한다.
2. mp의 정보를 vector<pair<string,int>>v에 저장한다.
3. customSort 함수 생성하기
   • 가장 자주 나오는 값 기준
   • 단어의 길이순
   • 오름차순
4. sort를 통해 정렬하기
5. v[i].first.length() ≥ m인 경우만 출력

✏️ 코드

#include <bits/stdc++.h>
using namespace std;

map<string, int> mp;

// customSort 함수 수정
bool customSort(const pair<string, int> &a, const pair<string, int> &b) {
    if (a.second != b.second) { // 가장 자주 나오는 값을 기준
        return a.second > b.second;
    }
    if (a.first.size() != b.first.size()) { // 단어의 길이순
        return a.first.size() > b.first.size();
    }
    return a.first < b.first; // 사전순 (오름차순)
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m;
    cin >> n >> m;

    for (int i = 0; i < n; i++) {
        string s;
        cin >> s;
        if (mp.find(s) == mp.end()) {
            mp.insert({s, 1});
        } else {
            mp[s] += 1;
        }
    }

    // map 데이터를 vector로 변환
    vector<pair<string, int>> v(mp.begin(), mp.end());
    // customSort를 사용해 정렬
    sort(v.begin(), v.end(), customSort);

    for (int i = 0; i < v.size(); i++) {
        if (v[i].first.length()>=m)
            cout<<v[i].first<<'\\n';
    }
}

 

✏️ 느낀 점

푸는데 얼마 안 걸린 것 같다. 처음엔 map을 통해서 하려고 했으나, vector에 저장하고 customSort를 이용해야겠다는 생각이 들었다.다만 customSort함수에서 부등호 방향이 헷갈려 살짝 헤멘 것 같다. 항상 기억하자 >가 오름차순

✏️ 결과