[python] 백준 1181 :: 단어 정렬

[단어 정렬]

# 문제
알파벳 소문자로 이루어진 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)]