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

문제

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

1181번: 단어 정렬

답

시행착오

import sys
input = sys.stdin.readline

N = int(input())
A = list(set(input() for _ in range(N)))

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

for i in A:
  print(i)

'출력 형식이 잘못되었습니다'

(P) sys.stdin.readline()은 '\n'을 포함하는 입력이기 때문에 연속으로 값을 입력받는 for문에서 에러가 발생했다.

(S) sys.stdin.readline().strip()

https://velog.io/@1204jh/1181

답

import sys
input = sys.stdin.readline

N = int(input())
A = list(set(input().strip() for _ in range(N)))

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

for i in A:
  print(i)

설명

1. 길이가 짧은 것부터 ㅡ> sort()

2. 길이가 같으면 사전 순으로 ㅡ> sort(key=len)

3. 중복은 제거 ㅡ> set() 사용