CodingTest/Baekjoon

[baekjoon] 백준 1181번(파이썬): 단어 정렬

JunJangE 2021. 12. 24. 14:21

문제

 

1181번: 단어 정렬

첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.

www.acmicpc.net

알고리즘

- set 자료형을 통해 중복을 제거하고 정렬을 위해 list 자료형으로 감싸준다.

- 길이가 짧은 것부터 길이가 같으면 사전 순으로 정렬하기 위해 sort() 함수를 사용한다.

- sort()함수를 통해 오름차순으로 정렬 후 단어의 길이를 기준으로 다시 오름차순으로 정렬한다.

- 반복문을 통해 단어를 출력한다.

코드

import sys

n = int(sys.stdin.readline())
# set 자료형을 통해 중복을 제거하고 정렬을 위해 list 자료형으로 감싸준다.
word = list(set(str(sys.stdin.readline().strip()) for _ in range(n)))
word.sort() # 오름차순 정렬
word.sort(key=lambda x: len(x)) # 단어의 길이를 기준으로 오름차순 정렬

# 반복문을 통해 단어를 출력
for i in word:
    print(i)

github

 

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법

내가 푼 코딩 테스트 문제와 해결법. Contribute to junjange/CodingTest development by creating an account on GitHub.

github.com