[python] 백준 1181 :: 단어 정렬
2023. 3. 8. 00:19ㆍAlgorithm
728x90
반응형
[단어 정렬]
# 문제
알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
길이가 짧은 것부터
길이가 같으면 사전 순으로
단, 중복된 단어는 하나만 남기고 제거해야 한다.
# 입력
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
# 출력
조건에 따라 정렬하여 단어들을 출력한다.
풀이
전체 코드
import sys
words = [sys.stdin.readline().strip() for _ in range(int(input()))]
def func(w):
return len(w), w
[print(w) for w in sorted(set(words), key=func)]
이 문제에는 정렬 조건이 두 개가 있다.
1) 길이가 짧은 것 우선
2) 사전 순
그리고 중복을 제거해야 한다.
python의 sort
혹은 sorted
를 써서 정렬할 수 있는데
여기서는 sorted
가 더 유용해 보인다.
👉🏻 왜냐하면 중복을 제거하기 위해 set
으로 바꿀 거라서 ㅎㅎ
list.sort() VS sorted(iterable) 비교
list.sort()
값을 반환하지 않고, 원본 배열을 수정한다.
sort()
는 list에만 사용할 수 있다.set, tuple, dict
에는 사용할 수 없다.
sorted(iterable)
원본 배열은 수정하지 않고, 정렬된 list를 반환해준다.
sorted(iterable)
은 list 이외에도 iterable이라면 모두 사용할 수 있다.set, tuple, dict
모두 사용 가능!!
_tuple = (2, 5, 4, 3, 1)
_dict = {2, 5, 4, 3, 1}
_set = set([2, 5, 4, 3, 1])
sorted_tuple = sorted(_tuple)
sorted_dict = sorted(_dict)
sorted_set = sorted(_set)
print(sorted_tuple) # [1, 2, 3, 4, 5]
print(sorted_dict) # [1, 2, 3, 4, 5]
print(sorted_set) # [1, 2, 3, 4, 5]
print(_tuple) # (2, 5, 4, 3, 1)
print(_dict) # {1, 2, 3, 4, 5}
print(_set) # {1, 2, 3, 4, 5}
이제 문제를 풀어봅시다
1. 중복을 제거한다.
set(words)
set(iterable)
만 해주면 중복이 제거된 set이 반환된다.
2. 정렬을 커스텀할 함수를 만든다.
# 정렬 커스텀 함수
def func(w):
# 길이순, 사전순
return len(w), w
sorted(set(words), key=func)]
sort와 sorted 모두 인자로
key=함수
를 보내주면, 해당 함수의 조건대로 정렬해준다.sorted 안에 정렬할 iterable과 커스텀 함수를 순서대로 넣어주면 된다.
3. 출력하면 끝~
[print(w) for w in sorted(set(words), key=func)]
- comprehension을 이용해 바로 출력해버림
전체 코드
import sys
words = [sys.stdin.readline().strip() for _ in range(int(input()))]
def func(w):
return len(w), w
[print(w) for w in sorted(set(words), key=func)]
728x90
반응형
'Algorithm' 카테고리의 다른 글
[python] 백준 2309 :: 일곱 난쟁이 (0) | 2023.03.08 |
---|---|
[python] 백준 10989 :: 수 정렬하기 3 (도수 정렬, 계수 정렬) (0) | 2023.03.07 |
[python] 백준 1074 :: Z (0) | 2023.03.07 |