[Python/코테] 백준 1181 단어 정렬

 

문제 및 조건 설명: https://www.acmicpc.net/problem/1181

알고리즘 설계

💡Idea.

• 정렬이 필요할 때는 표준 라이브러리를 사용한다.
• 중복 불가 조건을 만족시키기 위해 사전 자료형을 사용한다.

알고리즘 구현

출처: https://www.acmicpc.net/status?user_id=choiyn33&problem_id=1181&from_mine=1

import sys

n = int(input())
arr = []

for i in range(n):
    arr.append(sys.stdin.readline().strip())

set_arr = set(arr)  # 중복 단어 제거
arr = list(set_arr)  # 다시 list로 변환
arr.sort()
arr.sort(key=len)

for i in arr:
    print(i)

나다어

• 중복 제거를 위해 사전형을 사용한다
• arr.sort(key=len)은 길이를 기준으로 정렬해준다. sort 함수의 기능을 잘 알고 있자!