[Python] 백준 1181 단어 정렬 | 내장함수

~실버 5~

내장 타파!

내장 파괴!

 

 

문제 해결의 포인트( 내장 함수 )

- input 함수는 시간이 짱 많이 걸린다

- set 함수로 중복 제거

- sort 함수 커스터마이징

---

1. 시간 줄이기 | 시간 초과 피하기 | input 대신 readline

import sys

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

for _ in range(n):
	words.append(sys.stdin.readline())

sys 모듈을 import하여 sys.stdin.readline()을 사용하면 input()보다 시간을 확 줄일 수 있습니다..!

 

처음에 입력을 받을 때 리스트의 크기를 설정하고 (ex. list[[0]*50 for i in range(20000)]) 했는데 완전히 허튼짓이었다.

파이썬이 아닌 언어를 사용할 때 항상 자료구조의 크기나 변수의 타입을 명시하는 버릇이 남아있는 듯하다

 

 

2. set 함수로 중복 제거

set_words = set(words)
words = list(set_words)

자료구조에는 python에서 많이 사용하는 LIST뿐만 아니라 SET도 있는데 이는 중복 값을 허용하지 않아서 list를 set화 하게 되면 자동으로 중복값이 제거되어 변수로 저장한다.

 

이를 sort함수로 처리(정렬) 하기 위해서 다시 list형으로 변환하였다.

 

3. Sort 함수 커스터마이징

words.sort()
words.sort(key = len)

for i in words:
	print(i)

list.sort()사용하여 str타입의 값들을 먼저 정렬한다.

 

사전순으로 정렬 후 길이가 작은 것부터 다시 정렬한다.

이때 sort의 옵션 중 key를 사용하는데, function을 넘겨주면 된다(길이 오름차순이므로 len)

 

sort에 대한 더 자세한 내용은 여기를 클릭

 

 

 

<최종 코드>

import sys

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

for _ in range(n):
    words.append(sys.stdin.readline().strip()) #strip() : white space 제거

set_words = set(words)
words = list(set_words)

words.sort()
words.sort(key=len)

for i in words:
    print(i)

 

 

---

 

 

 

참조한 사이트 & 본인 Notion 페이지

https://velog.io/@1204jh/1181

[Python / 파이썬] 백준 1181번

https://sin-young.notion.site/E1181-af14b5381c8b4681b007656dc6a1331b

E1181