본문 바로가기
Programming/알고리즘

[백준 / Python] 1181번 단어 정렬

by castberry_ 2023. 7. 9.
반응형

문제

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

 

1181번: 단어 정렬

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

www.acmicpc.net


풀이 

 

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

for i in range(n):
    data.append(input())
data = set(data)
data = list(data)
data.sort(key = lambda x : (len(x), x))

for i in data:
    print(i)

 

이번 문제는 문자열들이 주어졌을때 

1. 중복을 제거한다. 

2. 길이순으로 정렬한다. 

3. 길이가 같다면 사전순으로 정렬한다. 

위 과정을 거치고 출력을 해야한다. 

 

1. 중복 제거

data = set(data)

data = list(data)

 

list를 set으로 바꾸고 list로 바꾸어 중복을 제거하였다. 

 

2. 정렬

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

sort에 람다함수를 넣어 정렬 기준을 변경해주었다. 

 

x : (len(x), x)

에서 len(x)는 길이순으로 정렬하는 코드이고, x는 길이가 같다면 사전순으로 정렬하는 코드이다. 

 

반응형

댓글