본문 바로가기
004 컴퓨터과학/알고리즘

[백준 온라인저지] 정렬 알고리즘 / 1181. 단어 정렬

by Dallas 2023. 6. 4.
반응형

문제

 

예제

 

소스코드

시간 : 4596ms

import sys

n = int(sys.stdin.readline())

answer = []
strings = []
for x in range(n) :
    string = sys.stdin.readline().strip()
    strings.append(string)
    
strings.sort(key = lambda x : (len(x), x))

for x in strings :
    if x not in answer :
        answer.append(x)
        
for x in answer :
    print(x)

 

메모

  • sys.stdin.readline() 을 사용하면, '\n' 이 삽입된다. -> strip() 으로 제거
  • 문자열 또한 sort()를 이용해 정렬할 수 있다.
  • sort(key = ``)를 통해 정렬 기준을 정할 수 있다.
반응형

댓글