< 백준 BaekJoon : 20920번 영단어 암기는 괴로워 > C++

<< 문제 >>

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

20920번: 영단어 암기는 괴로워

 

map을 활용해서 first에 단어, second에 등장 횟수를 저장합니다.

 

이렇게 모든 단어를 확인한 후,

 

map은 이미 순서가 오름차순으로 정렬되어 있기 때문에

 

vector<pair>로 이동시켜 새로 정렬한다. 

 

이 때, sort()에 새로운 comp 함수를 정의해서 풀이한다.

 

 

<< 코드 >> 

 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <map>
using namespace std;

bool comp(pair<string, int> p1, pair<string, int> p2){
  // 단어의 출현 수
  if(p1.second != p2.second){
      return p1.second > p2.second;
  }
  
  // 단어 길이
  if(p1.first.size() != p2.first.size()){
      return p1.first.size() > p2.first.size();
  }
  
  // 사전 순
  if(p1.first != p2.first){
      return p1.first < p2.first;
  }
}

int main(){
  ios_base :: sync_with_stdio(false);
  cin.tie(NULL);
  
  int numN, numM;
  cin >> numN >> numM;

  int top = 0;
  map<string, int> wordMap;
  string str;
  for(int i=0; i<numN; i++){
    cin >> str;
    if(str.size() < numM){
      continue;
    }
    
    auto address = wordMap.find(str);
    if(address == wordMap.end()){
      wordMap.insert({str, 1});
    }
    else{
      address->second++;
      if(address->second > top){
        top = address->second;
      }
    }
  }
  
  vector<pair<string,int>> vec;
  for(auto& a : wordMap){
    vec.push_back(a);
  }
  
  sort(vec.begin(), vec.end(), comp);
  
  for(auto& v : vec){
    cout << v.first << '\n';
  }
}

 

 

<< 깃헙 >>

https://github.com/J2on/StudyAlgorithm_Part2/tree/main/%EB%B0%B1%EC%A4%80/Silver/20920.%E2%80%85%EC%98%81%EB%8B%A8%EC%96%B4%E2%80%85%EC%95%94%EA%B8%B0%EB%8A%94%E2%80%85%EA%B4%B4%EB%A1%9C%EC%9B%8C