자료구조+알고리즘
백준-1181 단어 정렬하기 (sort 함수, lambda)
덕구공
2021. 5. 13. 02:03
문제
알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
- 길이가 짧은 것부터
- 길이가 같으면 사전 순으로
풀이
sort 함수의 key function에 lamba를 이용해서 길이순, 길이가 같으면 사전순으로 정렬하자 .
num = int(input())
words = []
for i in range(0, num):
new_words = input()
if (len(new_words), new_words) not in words:
words.append((len(new_words), new_words))
words.sort(key=lambda x: (x[0], x[1])) //길이순, 길이가 같으면 사전순으로
for i in words:
print(i[1])