< 백준 - 1181 > : 단어 정렬하기 


문제 : 
알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.
  1. 길이가 짧은 것부터
  2. 길이가 같으면 사전 순으로

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

출력 : 
조건에 따라 정렬하여 단어들을 출력한다. 단, 같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다.

예제 입력:

13
but
i
wont
hesitate
no
more
no
more
it
cannot
wait
im
yours

예제 출력 : 

i
im
it
no
but
more
wait
wont
yours
cannot
hesitate

 < 내 코드 > 코드 길이 : 252B 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
import sys
N = int(input())
list = []
for i in range(N):
    list.append(sys.stdin.readline())
my_set = set(list)
new_list = []
for i in my_set:
    new_list.append(i)
new_list.sort(key = lambda x : (len(x), x))

for i in new_list:
    print(i, end='')

문제의 '같은 단어가 여러 번 입력된 경우에는 한 번씩만 출력한다' 요구사항을 충족하기 위해 리스트를 집합으로 만들고(6번 줄) , 다시 그 집합을 리스트(7~9번 줄)로 만들었다.

좀 복잡하게 푼 것 같아 다른 사람의 코드를 찾아봤는데 훨씬 짧은 코드로 구현 가능하다는 것을 알 수 있다.

< 코드 길이 165B인 다른 코더의 코드 > 


1
2
3
4
5
6
7
8
9
N = int(input())
l = []
for _ in range(N):
    k = input()
    if k not in l:
        l.append(k)
        
for i in sorted(l, key=lambda x: (len(x),x)):
    print(i)
코드 출처 : ishuca