영단어 암기는 괴로워
문제
화은이는 이번 영어 시험에서 틀린 문제를 바탕으로 영어 단어 암기를 하려고 한다. 그 과정에서 효율적으로 영어 단어를 외우기 위해 영어 단어장을 만들려 하고 있다. 화은이가 만들고자 하는 단어장의 단어 순서는 다음과 같은 우선순위를 차례로 적용하여 만들어진다.
- 자주 나오는 단어일수록 앞에 배치한다.
- 해당 단어의 길이가 길수록 앞에 배치한다.
- 알파벳 사전 순으로 앞에 있는 단어일수록 앞에 배치한다
M보다 짧은 길이의 단어의 경우 읽는 것만으로도 외울 수 있기 때문에 길이가 M이상인 단어들만 외운다고 한다. 화은이가 괴로운 영단어 암기를 효율적으로 할 수 있도록 단어장을 만들어 주자.
입력
첫째 줄에는 영어 지문에 나오는 단어의 개수 N과 외울 단어의 길이 기준이 되는 M이 공백으로 구분되어 주어진다.
(1 ≤ N ≤ 100,000, 1 ≤ M ≤ 10)
둘째 줄부터 N+1번째 줄까지 외울 단어를 입력받는다. 이때의 입력은 알파벳 소문자로만 주어지며 단어의 길이는 10을 넘지 않는다.
단어장에 단어가 반드시 1개 이상 존재하는 입력만 주어진다.
출력
화은이의 단어장에 들어 있는 단어를 단어장의 앞에 위치한 단어부터 한 줄에 한 단어씩 순서대로 출력한다.
예제 입력 1
7 4
apple
ant
sand
apple
append
sand
sand
예제 출력 1
sand
apple
append
예제 입력 2
12 5
appearance
append
attendance
swim
swift
swift
swift
mouse
wallet
mouse
ice
age
예제 출력 2
swift
mouse
appearance
attendance
append
wallet
해당 문제는 string 문자열을 조건에 따라 정렬하는 문제이다.
#include <iostream>
#include <string>
#include <algorithm>
#include <vector>
#include <unordered_map>
using namespace std;
bool Comp(const pair<string, int> a, pair<string, int> b)
{
if (a.second == b.second)
{
if (a.first.length() == b.first.length())
return a.first < b.first;
else
return a.first.length() > b.first.length();
}
else
return a.second > b.second;
}
int main()
{
ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
int N, M;
cin >> N >> M;
unordered_map<string, int> inputs;
while(N--)
{
string input;
cin >> input;
if (input.length() >= M)
inputs[input]++;
}
vector<pair<string, int>> words(inputs.begin(), inputs.end());
sort(words.begin(), words.end(), Comp);
for (pair<string, int> word : words)
cout << word.first << '\n';
return 0;
}
자주 나오는 단어의 빈도수를 체크해야 하기 때문에 unordered_map 변수를 선언하여, 입력된 문자열을 key값으로 하고 카운팅 횟수를 value로 하여 초기화해주었다.
이후, 만들어진 unordered_map 변수를 vector로 다시 만들면서, Comp 함수를 통한 sort를 진행하였다.
bool Comp 함수는 문제의 조건에 따라 우선순위를 결정해주었다.
1. 자주 나오는 단어일수록 앞 순위이므로, pair<string, int>의 second 값을 비교하여 같은 경우엔 다음 조건을, 다른 경우엔 더 큰 second 값을 반환해주었다.
2. 다음으로 문자열의 길이를 비교하여 길이가 같은 경우엔 문자의 알파벳 사전순으로 정렬되도록 a.first < b.first를 수행하였고, 길이가 다른 경우엔 길이가 더 큰 값으로 비교하여 반환해주었다.
해당 문제를 만들면서 직면한 난관은 Comp 함수의 작성 중 발생하는 오류(invalid comparator)였다.
해당 오류는 "sort를 위한 compare 함수는 반드시 strict weak ordering(엄격한 약한 순서)을 만족해야 합니다."라는 의미인데,
이는 즉, a==b인 경우에는 a < b도 false이고, a > b도 false가 되도록 결과가 나와야한다는 뜻이다.
즉, 모든 경우의 수를 다 비교하여 올바른 return 값이 반환되도록 정렬함수를 작성하지 않으면 나타나는 오류이다.
https://www.acmicpc.net/problem/20920
20920번: 영단어 암기는 괴로워
첫째 줄에는 영어 지문에 나오는 단어의 개수 $N$과 외울 단어의 길이 기준이 되는 $M$이 공백으로 구분되어 주어진다. ($1 \leq N \leq 100\,000$, $1 \leq M \leq 10$) 둘째 줄부터 $N+1$번째 줄까지 외울 단
www.acmicpc.net
'💻코딩테스트 > 백준' 카테고리의 다른 글
[백준/C++] 12865번 : 평범한 배낭 (DP 문제, 0-1 배낭 문제, 0-1 Knapsack) (0) | 2023.07.25 |
---|---|
[백준/C++] 2293번 : 동전 1 (DP 문제, 경우의 수) (0) | 2023.07.24 |
[백준/C++] 2108번 : 통계학 (반올림, 최빈값 찾기) (0) | 2023.07.24 |
[백준/C++] 11053번 : 가장 긴 증가하는 부분 수열 (LIS 수열 문제) (0) | 2023.07.21 |
[백준/C++] 18405번 : 경쟁적 전염 (BFS 연습) (0) | 2023.07.19 |