[문제]
알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
- 길이가 짧은 것부터
- 길이가 같으면 사전 순으로
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]
'알고리즘 > 알고리즘 문제' 카테고리의 다른 글
[python] 백준 > DP > 포도주 시식 (0) | 2021.02.09 |
---|---|
[python] 백준 > DP > 연속합(1912) (0) | 2021.02.09 |
[python] 백준 > 이분 탐색 > 숫자 카드(10815) (0) | 2021.02.04 |
[python] 백준 > 이분 탐색 > 두 용액(2470) (2) | 2021.02.04 |
[python] 백준 > 그리디 > 단어 수학(1339) (0) | 2021.02.04 |