[백준 1181] 단어 정렬 - python

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

1181번: 단어 정렬

 

문제 풀이

• 길이순과 사전순을 이용한 정렬을 해야했다
• 둘을 이용하려면 lambda를 사용한 sort함수를 이용하면 된다 (이 문제에서 핵심)
• 처음에 중복을 제거하여 받기 위해 set을 이용해 word를 입력받았다
• words에는 (word길이, word)를 담아주었다 둘을 이용해 정렬을 해야하기 때문에
• 후에 list로 바꿔준 뒤 lambda를 이용해 길이와 사전 순으로 정렬했다
• lambda를 이용해 길이순으로 정렬만 원한다면 (key=lambda : len) 으로만 짜도 된다 처음에 이렇게 짰다가 틀림 ㅠ

n = int(input())
words = set()
for _ in range(n):
    word = input()
    words.add((len(word), word))
    
words = list(words)
# 길이가 짧은 것부터
# 길이가 같으면 사전순으로
words.sort(key = lambda x : (x[0], x[1])) # 알아두기

for w in words:
    print(w[1])