728x90
반응형
문제 출처:https://www.acmicpc.net/problem/1181
1181번: 단어 정렬
첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
www.acmicpc.net
풀이:
import sys
N = int(sys.stdin.readline())
array = []
for i in range(N):
array.append(sys.stdin.readline().rstrip())
set = set(array) #중복값을 제거
array = list(set)
array.sort() # 일단 사전순으로 정렬 해주고
array.sort(key=len) # 사전순으로 정렬된 array 를 길이순으로 정렬한다
for i in array:
print(i)
rs.strip 을 안해주면 오른쪽 \n도 같이 받아온다. 그러므로 rs.strip 으로 오른쪽 공백 제거
array 에 들어간 리스트를 set으로 중복으로 들어간 값을 제거해준다.
array.sort(key=len)
# array.sort(key=lambda x:len(x)) 로도 가능하다
이 문제도 한 줄로 푼 사람이 있어 첨부한다...
print(''.join(sorted(sorted(set(open(0).readlines()[1:])), key=len)))
# 백준 알고리즘 alphahoon님 소스
반응형
댓글