1181 단어 정렬

1181번: 단어 정렬 (acmicpc.net)

1181번: 단어 정렬

N = int(input())
lst = set()
lst2 = []
lst3 = []
max = 0
for i in range(N):
    lst.add(input())
for i in lst:
    lst2.append(i)
lst2.sort()
for i in range(20001):
    for j in lst2:
        if len(j) == i:
            lst3.append(j)
    if len(lst3) == len(lst):
        break
print(*lst3, sep='\n')

진짜 오랜만에 하나도 안 보고 맞은 문제..

문자들을 문자 길이대로 정렬 후 같은 길이면 알파벳 순으로 정렬하는 문제였는데

우선 sort로 알파벳순으로 정렬하고 그 다음에 가장 문자길이가 큰것을 기준으로해서 그때그때 해당 문자길이가 있는 만큼 for문을 돌려서 출력해야했다..

그런데 보니까 input에 중복이 포함되어있어서 이것을 삭제해야했고.. 이를 위해 set로 선언이 필요했었다..

set는 또 len같은 것을 못쓰고 add로 요소 추가를 해야함

또한, 한번에 배열을 출력하는 방법도 깨달았다. print(*lst3, sep='\n')이런식. 

 

그런데 더 간단한 방법이 있다고 한다.....

import sys

n=int(sys.stdin.readline())
lis=[]
for _ in range(n):
    words=input()
    lis.append(words)
lis=list(set(lis))

lis.sort()
lis.sort(key=len)

for i in lis:

    
    print(i)

그냥 lst.sort(key=len)하면 되는거였음 ㅋ

파이썬엔 별게 다있네..