정렬: 백준 1181 파이썬

문제: 

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

1181번: 단어 정렬

 

정답:

import sys

n = int(sys.stdin.readline())

array = [input() for _ in range(n)]

# 입력받은 문자열의 중복을 제거하기 위해 set으로 바꾸었다가 다시 list로 변경
array = set(array)
array = list(array)

# 문자열 길이와 문자를 기준으로 정렬
array.sort(key=lambda x: (len(x), x)) 

for i in array:
    print(i)

 

풀이:

sort 함수와 sorted 함수를 사용할 때 key를 잘 사용하면 원하는 대로 정렬을 아주 쉽게 할 수 있다.