SWEA: 7701 염라대왕의 이름 정렬

문제

SW Expert Academy

 

문제 풀이

 이름이 주어질 때 다음과 같은 조건을 통해 정렬 후에 이름을 출력하면 되는 문제이다.

• 동일한 이름은 하나만 출력한다.
• 이름의 길이를 우선으로 정렬한다.
• 길이가 같다면, 이름 순으로 정렬한다.

위는 파이썬에서 `set`과 `lambda`를 활용하면 쉽게 해결 할 수 있다.

 

코드

T = int(input())

for test_case in range(1, T + 1):
    N = int(input())
    names = [None] * N

    for idx in range(N):
        cur_name = input()
        names[idx] = (cur_name, len(cur_name))

    # set으로 중복 제거, 이름의 길이가 우선이 되도록 정렬
    result = sorted(set(names), key=lambda x: (x[1], x[0]))
    print('#{}'.format(T))
    print(*[name for name, _ in result], sep='\n')