단어장 만들기

Last updated Dec 14, 2022 Edit Source

• Lv 2
• 구현

1
2
3
4
5
6
7
8

7 3 
aaaa 
aaa 
aa 
a 
b 
bb 
bbb

• 단어들을 정렬하고 $k$ 번째에 있는 단어 출력하기
• 정렬은 길이 순서
• 단, 길이가 같은 단어의 경우 사전 순서

1
2
3
4
5
6
7
8

user_input = input()
n, k = map(int, user_input.split())
ls = []
for _ in range(n):
	ls.append(input())

ls.sort(key=lambda x: (len(x), x))
print(ls[k-1])

Note

조건이 많이 나오는 유형이라 기록한다. 이 문제는 리스트에 존재하는 문자들을 길이와 사전 순으로 정렬하는 문제이다. 문자열 구현 유형에서 많이 나오는 스킬인 만큼 ls.sort(key=lambda x: (len(x), x)) 테크닉은 익혀두는 것이 좋다. 그냥 key=len으로 하면 길이순으로만 요소를 정렬하지만 key=lambda x: (len(x), x)으로 하면 길이와 사전 순으로 정렬시킨다.