[파이썬] 백준 1181번 단어 정렬

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

1181번: 단어 정렬

[문제 이해]

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래의 조건에 따라 정렬하는 프로그램을 작성하는 문제이다.

1. 길이가 짧은 것부터

2. 길이가 같으면 사전 순으로

단, 중복된 단어는 하나만 남기고 제거한다.

 

[문제 풀이]

단어 리스트를 받았을 때, 조건대로 짧은 순으로 먼저 정렬하고, 만약 길이가 같으면 사전 순으로 정렬하는 문제라 어렵지 않게 풀 수 있었다. 그 전에 받은 리스트 중 중복된 단어는 하나만 남기고 제거하기 때문에 set()함수를 사용하여 중복을 제거해주고, lambda를 이용하여 sorted() 내에 key 값을 어떤 것을 기준으로 할지 정하여 구현하면 조건에 맞는 단어 리스트 순서대로 출력할 수 있다.

import sys

N = int(sys.stdin.readline()) # 단어 개수
str_list = list() # 입력받을 단어 리스트
for i in range(N):
    str_list.append(sys.stdin.readline().strip()) 
    # strip() 사용하여 각 원소의 '\n' 삭제

str_list = set(str_list) # 중복 제거
str_list = sorted(str_list, key = lambda x: (len(x), x)) 
# 길이가 짧은 것부터 정렬하고 만약 길이가 동일하면 x를 기준으로 정렬 = 사전순 정렬

for i in str_list:
    print(i)

 

 

lambda를 사용하여 x의 길이인 len()함수를 기준으로 먼저 정렬하고, 그 다음에는 x를 넣어 사전순으로 정렬될 수 있도록 한다.