728x90
반응형
문제
알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
- 길이가 짧은 것부터
- 길이가 같으면 사전 순으로
입력
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
출력
조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.
예제 입력 1
13
but
i
wont
hesitate
no
more
no
more
it
cannot
wait
im
yours
예제 출력 1
i
im
it
no
but
more
wait
wont
yours
cannot
hesitate
[문제 해설]
이 문제는 두 가지 해결해야 될 점이 있다. 먼저 길이가 짧은 순서대로 단어를 정렬해야 되고, 두 번째 그 정렬된 상태를 단어 사전 순서대로 정렬하면 된다. 두 가지 조건이 있으면 lambda 함수를 떠올려서 정렬하는 방법을 생각하면 된다.
그리고 중복을 제거해야 하기 때문에 set으로 바꾼뒤에 다시 list형식으로 바꾸어 준다.
n = int(input())
datas=[]
for _ in range(n):
word = input()
word_c = len(word)
datas.append((word,word_c))
#중복 값 제거
datas=list(set(datas))
datas.sort(key=lambda word:(word[1],word[0]))
for i in range(len(datas)):
print(datas[i][0])
728x90
반응형
'알고리즘 > 알고리즘 문제' 카테고리의 다른 글
[파이썬] [백준 -1026번] 보물 (Dynamic Programming) (0) | 2021.03.29 |
---|---|
[파이썬] [백준 - 10814번] 나이순 정렬 (Sort) (0) | 2021.03.29 |
[파이썬] [백준 - 1931번] 회의실 배정 (Dynamic Programming) (0) | 2021.03.29 |
[파이썬] [백준 - 10989번] 수 정렬하기 3 (Sort) (0) | 2021.03.29 |
[파이썬] [백준 - 14501번] 퇴사 (Dynamic Programming) (1) | 2021.03.29 |