자료구조+알고리즘

백준-1181 단어 정렬하기 (sort 함수, lambda)

덕구공 2021. 5. 13. 02:03

문제

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.

  1. 길이가 짧은 것부터
  2. 길이가 같으면 사전 순으로

풀이

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])