1181 집합 원소 길이별 정렬

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

1181번: 단어 정렬

 

흠.. 사실 중복을 제거해야되는부분을 못보고 리스트로 할생각이였다. 내가 궁금했던건, 리스트를 len(요소)에 따라 sort가 가능한가? 였는데 파이썬은 물론 가능했다.

 

N = int(input())
words = set()

for _ in range(N):
    words.add(input().strip())

# 길이가 짧은 것부터, 길이가 같으면 사전 순으로 정렬
sorted_words = sorted(words, key=lambda x: (len(x), x))

for word in sorted_words:
    print(word)

 

리스트 또한 이런식으로 정렬할수가 있더라.. 이전에 공부했던 람다에 따라 첫 정렬 조건을 len(x), 두번째 조건을 x로 주고 정렬한것이다.

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

for _ in range(N):
    word = input().strip()
    words.append(word)

# 중복 제거 후 다시 리스트로 변환
words = list(set(words))

# 길이가 짧은 것부터, 길이가 같으면 사전 순으로 정렬
words.sort(key=lambda x: (len(x), x))

for word in words:
    print(word)

 

리스트 말고 집합으로 했어야했지만, 암튼 리스트도 가능하다는거. 두 방식은

sorted_words = sorted(words, key=lambda x: (len(x), x))

words.sort(key=lambda x: (len(x), x))

 

이렇게 차이가 나는구나