[Python Algorithm] 단어 정렬 BOJ #1181

문제

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

1181번: 단어 정렬

 

내 문제 풀이

import sys

n = int(sys.stdin.readline())
result = []

for _ in range(n):
    now = sys.stdin.readline().strip()
    result.append((len(now), now))

result = list(set(result))
result.sort()

for i in range(len(result)):
    print(result[i][1])

👉 문자열 문제로, set과 튜플, 그리고 sort를 알맞게 사용해주면 됨

• 입력이 아주 많이 있을 수 있으므로 input() 대신 sys.stdin.readline을 이용해주었고,
  입력을 문자열 그대로 사용할 예정이므로 맨 뒤에 붙은 '\n'을 삭제하기 위해 strip()을 붙였다.
• 입력받은 문자열은 (문자열의 길이, 문자열) 의 튜플 형태로 result 리스트에 모두 추가했다.
• 완성된 result 리스트는 set로 만들어서 중복을 제거하고, 다시 list로 만들어주었다.
• sort를 사용하여 정렬했는데, 앞서 말했듯 (문자열의 길이, 문자열)의 형태로 저장했기 때문에
  문자열의 길이를 우선으로 정렬되고, 길이가 같을 경우 문자열을 기준으로 정렬된다.

 

💡 피드백

• 쉬운 문제지만 문자열 문제를 너무 오랜만에 풀었더니 중복을 set로 제거하는 것을 까먹었다.
  그래서 처음에는 일일이 다 비교해주었더니, 시간초과가 발생했다.