개발 연습장/백준 문제풀이

[파이썬, Python] 백준 1181: 단어 정렬

LooanCheong 2023. 1. 8. 20:27
반응형

문제

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

 

1181번: 단어 정렬

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

www.acmicpc.net

코드

n = int(input())
text_list = []

for _ in range(n):
    text_list.append(input())

text_list = list(set(text_list))
text_list.sort()
text_list.sort(key=len)

for i in text_list:
    print(i)

설명

우선 단어들을 입력받고 한 번 정렬을 해준다.
이후 한번 더 sort를 해주는데 이때 key 값으로 len을 주어 단어의 길이 순서로 정렬한다.

반응형