[파이썬] 1181. 단어 정렬

문제

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

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

입력

첫째 줄에 단어의 개수 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

 

 

🥨 나의 풀이

 

string=set(input() for _ in range(int(input())))    # 중복 제거
string=sorted(string)  # 사전 순 정렬
string=sorted(string, key=lambda x : len(x))    # 길이 순 정렬
for i in string : print(i)

 

단어의 갯수만큼 반복하며 입력받은 문자열들을 set에 담아서 중복을 제거한다.

길이가 짧은 것부터, 길이가 같으면 사전 순으로 이 두 가지 조건을 맞추기 위해 먼저 sorted를 사용해서 사전 순 정렬.

key에 lambda를 사용해서 입력 값 x를 길이 순으로 정렬하도록 했다.

반복문으로 문자열들을 하나씩 출력한다.

 

 

💃 다른 사람 풀이

 

import sys
word=set()
for i in range(int(input())) :
  word.add(sys.stdin.readline().rstrip())
word=list(word)
word.sort()
word.sort(key=lambda x:len(x))
print("\n".join(word))

 

word에 set의 메소드 add로 입력값을 추가한다.

word를 list로 변환시키고 sort를 쓰는 대신 바로 sorted를 쓰면 좋았을 것 같다.

마지막 출력 부분을 보니 join을 사용하고 줄바꿈을 하며 출력한다.

좋았던 부분 print("\n".join(word))

 

 

 

문제 출처 💁‍♀️ 백준