14-wkdghdwns199

[wkdghdwns199]

🔗 문제 링크

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

문제
화은이는 이번 영어 시험에서 틀린 문제를 바탕으로 영어 단어 암기를 하려고 한다. 그 과정에서 효율적으로 영어 단어를 외우기 위해 영어 단어장을 만들려 하고 있다. 화은이가 만들고자 하는 단어장의 단어 순서는 다음과 같은 우선순위를 차례로 적용하여 만들어진다.

1. 자주 나오는 단어일수록 앞에 배치한다.
2. 해당 단어의 길이가 길수록 앞에 배치한다.
3. 알파벳 사전 순으로 앞에 있는 단어일수록 앞에 배치한다.

M보다 짧은 길이의 단어의 경우 읽는 것만으로도 외울 수 있기 때문에 길이가 $M$이상인 단어들만 외운다고 한다. 화은이가 괴로운 영단어 암기를 효율적으로 할 수 있도록 단어장을 만들어 주자.

입력
첫째 줄에는 영어 지문에 나오는 단어의 개수 $N$과 외울 단어의 길이 기준이 되는 $M$이 공백으로 구분되어 주어진다. ($1 \leq N \leq 100,000$, $1 \leq M \leq 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

✔️ 소요된 시간

30분 - dictionary 자료형의 우선순위 정렬을 확인해 볼 수 있었다..!

✨ 수도 코드

결국 들어온 단어들을 다음과 같은 조건들을 통해 입력받은 단어들을 정렬하라는 문제였습니다.

1. 자주 나오는 단어
2. 해당 단어의 길이가 길수록 앞에
3. 알파벳 사전 순으로 앞에 있는 단어일수록 앞에

list를 다 따로 따로 정렬해서 해야 할까 라는 생각에 막막했지만, dictionary 로 자주 나오는 단어를 key 값으로 삼아 sorted 함수로 우선순위 정렬을 할 수 있지 않을까라는 막연한 생각으로 다음과 같은 코드를 짰습니다.

# 통과된 코드
import sys
input = sys.stdin.readline
N,M = map(int, input().split())
word_count = {}
for _ in range(N):
    word = input().rstrip()
    if len(word) >= M :
        if word in word_count :
            word_count[word][0] += 1
        else :
            word_count[word] = [1,len(word)]


sorted_word_list = sorted(word_count, key=lambda key : (-word_count[key][0], -word_count[key][1], key))
print()
for word in sorted_word_list :
    print(word) 

되더라구요! 저번에 태원 님이 올리신 PR 에서 말씀하신 조건 정렬 방식을 응용해서 다중 조건을 걸 수도 있다는 것을 알 수 있는 문제였습니다!

📚 새롭게 알게된 내용

[SeongHoonC]

kotlin 으로 푸는데 시간초과 너무 억까해서 포기..답은 잘 나옵니다.

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = System.out.bufferedWriter()
    val (N, M) = br.readLine().split(" ").map { it.toInt() }

    val dict = hashMapOf<String, Int>()
    for (i in 0 until N) {
        val word = br.readLine()!!
        if (word.length < M) {
            continue
        }
        dict[word] = (dict[word] ?: 0) + 1
    }
    dict.entries.sortedWith(compareBy({ -it.value }, { -it.key.length }, { it.key })).forEach {
        bw.write("${it.key}\n")
    }
    bw.flush()
    bw.close()
}

[alstjr7437]

단어 길이를 따로 배열에 정리를 해두셨군요!!

    else: # 단어가 M이상인 경우
        if word in word_lst: # 단어가 있는 경우
            word_lst[word] += 1
        else: # 단어가 없는 경우
            word_lst[word] = 1

word_lst = sorted(word_lst.items(), key = lambda x : (-x[1], -len(x[0]), x[0])) # x[0] = 단어, x[1] = 단어 빈도수

이렇게 넣어놓은 다음 두번째 순서로
딕셔너리에 key의 len이 긴거 먼저 정렬되게 정렬을 해줬습니다!!

[fnzksxl]

저는 정렬하는 부분은 민석님이랑 똑같은데요
딕셔너리에서 get(value, 없을때사용할 값) 으로 하면 값이 들어있지 않아도 쓸 수 있는걸로
코드를 좀 편하게 쓸 거 같아요. 홍준님처럼 길이까지 저장하시면 이 방법은 못 씁니다만..

for _ in range(N):
    word = input().rstrip()
    if len(word) >= M :
        word_count[word] = word_count.get(word, 0) + 1

p.s 파이썬에서 len 함수를 쓰면 시간복잡도는 어떻게 될까요? O(1) 입니다.
파이썬 객체는 길이가 변할 때마다 내부의 카운터가 증감하므로 이미 저장된 value를 꺼내오는 작업이라
길이를 반환하는 함수가 빠르게 돌아갑니다!