본문 바로가기
Backjoon

Backjoon problem 1181 단어 정렬 - python

by ddahu 2023. 5. 18.

https://www.acmicpc.net/problem/1181

 

1181번: 단어 정렬

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

www.acmicpc.net

  • 입력과 출력

  • 소스코드
lst = [input() for _ in range(int(input()))]
words = list(set(lst))
words.sort()
print("\n".join(map(str,sorted(words,key=lambda x:len(x)))))
  • 해설

이 문제는 sort 함수랑 set , key를 활용하여 푼 문제이다

우선 중복을 제거하기 위해 set으로 겹치는 중복 문자를 제거하고

sort()로 정렬한뒤 

정렬 된 sort words 리스트를 lambda 식을 통해 길이를 기준으로 정렬하여 푼 문제이다.