백준 1181번[단어 정렬] 문제풀이

1181 : 단어 정렬

📌문제 설명

BOJ 단어 정렬 - 1181

입력 예제

출력 예제

📌문제 접근 방식

1. 중복된 단어를 제거한다.

2. 글자 수로 정렬 후 알파벳 사전순으로 정렬한다.

 

📌몰랐던 점

sorting 기준을 lambda식으로 정의할 때 2가지 기준을 설정할 것

중복된 단어를 제거하는 건 set 자료구조를 사용해 간단히 해결할 수 있었다.

2단계에서 글자 수로 정렬하는 것도 문제없이 수행했으나

알파벳 사전순으로 정렬하는 과정에서 헷갈렸던 부분은 문자열 정렬을 어떻게 할 것인가였다.

lambda 식을 이용해 정렬할 때 2개의 기준을 주고 싶었는데 문법이 기억나지 않았다.

그래서 직접 문자열 정렬을 하는 함수로 구현하다보니 시간이 오래걸렸다.

sort함수의 key파라미터로 lambda 식을 넘겨줄 때 글자수와 문자열을 설정하면 간단한 문제였다.

📌소스코드 

import sys

n = int(sys.stdin.readline())
words = [sys.stdin.readline().strip() for i in range(n)]

def solution(words):
    result = list(set(words))
    result.sort(key = lambda word : (len(word), word))
    return result

result = solution(words)
for answer in result:
    print(answer)