[백준/Python] 단어 정렬

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

1181번: 단어 정렬

단어 길이 기준으로 정렬하는 조건을 넣어 정렬하면 길이 순으로 정렬되고 같은 길이 안에서는 사전 순으로 정렬된다고 생각했다. 별다른 조건없이 정렬했을 때 사전 순으로 정렬되니까! 근데 결과를 출력해보니 같은 길이 안에서 사전 순으로 정렬되지 않았다... ㅜ

import sys

n = int(sys.stdin.readline())
words = []

for _ in range(n):
	words.append(sys.stdin.readline().strip())

words.sort(key = len)

for word in words:
	print(word)

 

그래서 단어의 길이를 리스트에 저장해두고 먼저 단어 길이 짧은 순으로 정렬한 후 사전 순으로 정렬하도록 바꾸었다. 문제에서 같은 단어는 한 번만 출력된다는 조건이 있으니까 꼭 중복 제거를 해줘야 한다!

import sys

n = int(sys.stdin.readline()) # 단어 개수
words = [] # 단어 리스트

for _ in range(n):
	word = sys.stdin.readline().strip()
	words.append((word, len(word)))

words = list(set(words)) # 중복 제거
words.sort(key = lambda x: (x[1], x[0])) # 1. 단어 길이 짧은 순 2. 사전 순 으로 정렬

for word in words:
	print(word[0])