[백준] boj 1181 단어정렬

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

1181번: 단어 정렬

처음에는 functools 이용, 다음과 같이 커스텀 정렬함수를 구현하여 sort 함수에 key 값으로 주었다.

import sys
import functools

input = sys.stdin.readline
words = list(set([input().rstrip() for _ in range(int(input()))]))


def compare(item1, item2):
    if len(item1) < len(item2):
        return -1
    elif len(item1) > len(item2):
        return 1

    for i in range(len(item1)):
        if (ord(item1[i]) - ord(item2[i])) < 0:
            return -1
        elif (ord(item1[i]) - ord(item2[i])) > 0:
            return 1
    return 0


result = sorted(words, key=functools.cmp_to_key(compare))

for w in result:
    print(w)

참고링크 - https://lucky516.tistory.com/4

파이썬 커스텀 정렬 이용하기

=> 리턴값이 1, 0, -1 세개여야함.

=> 오름차순으로 정렬하고 싶을때

if first < last:

    return -1 

elif first == last:

    return 0

else:

    return 1 

 

 

그런데 sort 내장함수가 문자열을 사전순으로 정렬해주고 키값으로 len 을 주면 길이가 짧은 순으로 정렬해준다.

따라서 다음과 같이 코딩하면 훨씬 효율적이다

import sys

input = sys.stdin.readline
words = list(set([input().rstrip() for _ in range(int(input()))]))

words.sort()
words.sort(key=len)

for w in words:
    print(w)