본문 바로가기

알고리즘/알고리즘 문제

[python] 백준 > 정렬 > 단어정렬(1181)

[문제]

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

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

www.acmicpc.net/problem/1181

 

1181번: 단어 정렬

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

www.acmicpc.net

[나의 풀이]

import sys
input = sys.stdin.readline

# 입력
N = int(input().rstrip())
arr = set(input().rstrip() for _ in range(N)) # 중복 제거

# 길이가 짧은 것부터, 길이가 같으면 사전 순으로
answer = sorted(sorted(arr), key=lambda x: len(x))

# 출력
[print(a) for a in answer]