코테 공부

백준[1181] 단어정렬 (파이썬 풀이)

moonsun623 2021. 10. 1. 19:46
반응형

https://www.acmicpc.net/problem/1181

 

1181번: 단어 정렬

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

www.acmicpc.net

 

내가 한 풀이

n=int(input())
words=[]

for i in range(n):
    words.append(input())
    
for i in range(0,n):
    words[i]=(words[i],len(words[i]))
words=set(words)

#이중조건으로 리스트 정렬(x[1]를 기준으로 오름차순 정렬후 x[0]를 기준으로 다시 오름차순 정렬)
#(-를 붙이면 내림차순으로 정렬하게 된다.)
sorted_words=sorted(words,key=lambda x:(x[1],x[0]))

for word,length in sorted_words:
    print(word)

리스트에 (단어, 단어의 글자수) 형식으로 원소를 추가했다.

그 후 중복 단어를 제거하기위해 set으로 변환 해주고 이중 조건을 사용하여 리스트를 정렬했다.