[백준 #1181] 단어정렬 (파이썬)

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

1181번: 단어 정렬

 

import sys

n = int(sys.stdin.readline().strip()) #n = int(input())

dic = {}
for _ in range(n) :
    word = sys.stdin.readline().strip() #word = input()
    l = len(word)
    if l in dic :   dic[l].append(word) # 이미 딕셔너리에 존재할 경우
    else :          dic[l] = [word] # 딕셔너리에 해당 길이인 단어가 없을 경우

key_list = sorted(list(dic.keys())) # 길이 순 정렬
for key in key_list:
    for w in sorted(list(set(dic[key]))):  # 같은 길이인 단어들 중 중복 제거 후, 정렬
        print(w)

 

 

간단한 string 문제이기 때문에 속도를 높이기 위해 노력했다

dictionary 사용 / key 기준 정렬로 전체 단어들 정렬 피했다. 

그 결과 pypy 제출 답안중에서는 괜찮은 속도인 것 같다

그래도 더 괜찮은 코드가 있을것같은데 잔디 채우고싶어서 술먹고꾸역꾸역 푼거라서 내일...찾아봐야지....