[백준] 1181번 - 단어 정렬 (정렬) - 결과 포함

www.acmicpc.net/problem/1181

1181번: 단어 정렬

문자열들을 입력 받아 주어진 조건에 맞게 정렬하는 문제이다.

이전에 좌표를 정렬했던 문제들과 유사한 문제였다.

 

두 가지 조건이 주어지고 이에 따라 정렬하면 되는 문제였다.

이전과 마찬가지로 sort() 함수의 key 파라미터에 두 가지 조건을 전달했다.

 

첫 번째 조건은 문자열의 길이였다.

len() 함수를 이용해서 현재 문자열의 길이를 전달했다.

 

만약 문자열의 길이가 동일한 원소들이 존재할 경우 두 번째 조건에 따라 정렬한다.

사전 순으로 정렬하는 것인데 파이썬에서는 문자열 자체를 기준으로 정렬하면

자동으로 사전 순으로 정렬한다. 그러므로 두 번째 조건으로 문자열 값 자체를 넘겨주었다.

 

[소스 코드]

import sys

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

data = set()

for i in range(n):
    data.add(sys.stdin.readline().rstrip())

data = list(data)
data.sort(key=lambda str : (len(str), str))

for str in data:
    print(str)