알고리즘/Baekjoon
[CodingTest] python 백준 Baekjoon #1181 단어 정렬 / 정렬
Happyoon ~
2021. 11. 19. 02:20
728x90
반응형
https://www.acmicpc.net/problem/1181
1181번: 단어 정렬
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
www.acmicpc.net
나의 코드
#입력받은 문자열 저장 리스트
arr = []
#입력받은 숫자만큼 문자열 입력받고,
for i in range(int(input())):
s = input()
#입력받은 문자열과 중복되는 문자열이 arr에 없으면
if s not in arr:
#arr에 문자열 삽입
arr.append(s)
#문자열의 길이 저장 배열
length = [len(item) for item in arr]
#zip으로 문자열과 길이 묶기
pairs = zip(arr,length)
#zip의 각 원소들을 list 형태로 바꾸고, 다시 list로 묶어서 이중 리스트 형성
lst = []
lst = list(map(list,pairs))
#길이=>알파벳 순으로 정렬
lst.sort(key = lambda x:(x[1],x[0]))
#문자열 출력
for item in lst:
print(item[0])
다른 사람 코드
import sys
word=set()
for i in range(int(input())):
word.add(sys.stdin.readline().rstrip())
word=list(word)
word.sort()
word.sort(key=lambda x:len(x))
print("\n".join(word))
이 분은 중복되지 않게 집합을 사용했는데 좋은 것 같다.
sort() 한번에 한건 내가 더 잘한듯 ㅎ (나만의 생각 ㅋ)
마지막 출력도 나는 for문을 썼는데 그냥 join으로 출력하면 시간복잡도를 줄일 수 있을 것 같다.
왠지 모르겠는데 다른분들 코드가 다 똑같다 .. 정석인가봄
반응형