[알고리즘] 백준 1181 파이썬 - 단어 정렬

 

1181번: 단어 정렬

문제

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

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

 

문제풀이

import sys
input = sys.stdin.readline
A=[]
for _ in range(int(input())):
    a = input().rstrip()
    if a not in A:
        A.append(a)
A.sort()
A.sort(key=len)
for i in A:
    print(i)

 

- 사전 순으로 정렬한 뒤 key를 이용해 길이가 짧은 순으로 정렬한다.

  역순으로 진행할 경우 결국 사전 순으로 정렬되어서 의미가 없어진다.