백준 1181번 풀이

문제

주어진 영단어를 주어진 순서대로 정렬하는 문제

풀이

자 일단 이 문제를 풀기 위해서는 총 두가지를 해야 한다. 중복되는 단어를 없앤 다음 주어진 순서대로 정렬하는 것. 리스트 출력은 별개로 둡시다…

엥? 중복 단어요? 그걸 어떻게 가려내죠? 사람이 일일이 인덱싱 해야 하나요? 에이 그런 노가다 안해도 됨. 파이썬에는 세트(set)라는 매우 좋은 형태가 있다. 세트는 안에 든 것이 중복되면 하나 빼고 다 날려버리기때문에 입력 받은걸 리스트로 저장하고, 그 리스트를 세트로 만든 다음 다시 리스트로 만들면 중복되는 단어들은 깔끔하게 해결된다.

import sys

N = int(sys.stdin.readline())
word_list = []

for i in range(N):
    word = sys.stdin.readline().strip()
    word_list.append(word)

word_list = set(word_list)
word_list = list(word_list)
# set으로 중복 날려버리기 신공

그래서 이렇게만 해주면 중복되는 단어들은 아웃! 이제 남은 것은 중복된 단어가 없어진 리스트에서 단어들을 정렬하는 일이다.

일단 문제에서 주어진 정렬 기준을 보자. 첫번째로 단어의 길이, 두번째로 ABC순으로 정렬되는데 이게 무슨 소리냐면 i는 한글자고 apple은 다섯글자니까 i가 apple보다 위에 온다는 얘기다. bead와 bean은 둘 다 네글자이므로 bead, bean 순으로 오게 된다. 아니 그럼 이걸 언제 짜죠? 아니 뭘 또 짜요 김람다씨 불러.

word_list.sort(key=lambda x:(len(x),x))
# 솔직히 김람다씨 될 줄 몰랐죠... 

일단 저기서 순서는 둘째치고… 엥? 괄호로 묶었어요? 람다식으로 정렬할 기준이 여러개면 튜플로 묶으면 된다. 즉, key=lambda x:(len(x),x)에서 (len(x), x)가 정렬 기준 순서. 저렇게 하면 일차적으로 단어의 길이순으로 정렬한 다음 다시 사전 순서대로 정렬하게 된다.

import sys

N = int(sys.stdin.readline())
word_list = []

for i in range(N):
    word = sys.stdin.readline().strip()
    word_list.append(word)

word_list = set(word_list)
word_list = list(word_list)
# set으로 중복 날려버리기 신공

word_list.sort(key=lambda x:(len(x),x))
# 솔직히 김람다씨 될 줄 몰랐죠... 

for i in word_list:
    print(i)

근데 이게 한번에 될 줄은 몰랐지…